Home
last modified time | relevance | path

Searched defs:first_input (Results 1 – 12 of 12) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dpack_op.cc51 const Tensor& first_input = c->input(0); in Compute() local
Dconcat_op.cc99 const Tensor& first_input = c->input(values_input_start_index_); in Compute() local
/external/tensorflow/tensorflow/core/common_runtime/
Dexecutor.cc514 Entry* first_input; member
594 const TaggedNode& tagged_node, Entry* first_input, in ProcessAsync()
604 Entry* first_input = state->first_input; // Shorthand in ProcessAsync() local
770 Entry* first_input = propagator_.GetInputTensors(tagged_node); in Process() local
854 const NodeItem& item, Entry* first_input, TensorValueVec* inputs, in PrepareInputs()
/external/tensorflow/tensorflow/core/ops/
Dcontrol_flow_ops.cc101 ShapeHandle first_input = c->input(1); in __anon92ff84320202() local
/external/tensorflow/tensorflow/core/kernels/data/
Dsingle_threaded_executor.cc334 Entry& first_input = inputs[arg_output_locations_[i][0]]; in Run() local
/external/mesa3d/src/gallium/auxiliary/vl/
Dvl_idct.c432 unsigned first_input, struct ureg_dst fragment) in vl_idct_stage2_frag_shader()
Dvl_mpeg12_decoder.c1089 unsigned first_input, in mc_frag_shader_callback()
/external/swiftshader/third_party/SPIRV-Tools/source/opt/
Dfolding_rules.cpp1654 Instruction* first_input = in VectorShuffleFeedingExtract() local
/external/deqp-deps/SPIRV-Tools/source/opt/
Dfolding_rules.cpp1654 Instruction* first_input = in VectorShuffleFeedingExtract() local
/external/angle/third_party/vulkan-deps/spirv-tools/src/source/opt/
Dfolding_rules.cpp1707 Instruction* first_input = in VectorShuffleFeedingExtract() local
/external/tensorflow/tensorflow/compiler/mlir/lite/ir/
Dtfl_ops.cc1498 Operation *first_input = pack_op.getOperand(0).getDefiningOp(); in matchAndRewrite() local
/external/tensorflow/tensorflow/core/grappler/optimizers/
Darithmetic_optimizer.cc1706 const int first_input = concat_node->op() == "Concat" ? 1 : 0; in HoistChainForConcat() local