Searched refs:kAuxInputTensor (Results 1 – 4 of 4) sorted by relevance
67 const bool hasAuxInput = !context->isOmittedInput(kAuxInputTensor); in getLinkingMode()116 auxInput = context->getInputBuffer<T>(kAuxInputTensor); in executeTyped()120 auxInput = context->getInputBuffer<T>(kAuxInputTensor); in executeTyped()125 Shape auxInputShape = context->getInputShape(kAuxInputTensor); in executeTyped()318 Shape auxInput = context->getInputShape(kAuxInputTensor); in prepare()
148 aux_input_ = GetInput(operation, operands, kAuxInputTensor); in BidirectionalSequenceLSTM()
37 constexpr uint32_t kAuxInputTensor = 9; // optional variable
117 static constexpr int kAuxInputTensor = 39; // Optional variable