| /external/tensorflow/tensorflow/core/kernels/ |
| D | pack_op.cc | 51 const Tensor& first_input = c->input(0); in Compute() local
53 int expanded_num_dims = first_input.dims() + 1; in Compute()
62 TensorShape output_shape(first_input.shape()); in Compute()
68 CHECK(output.CopyFrom(first_input, output_shape)); in Compute()
100 OP_REQUIRES(c, first_input.shape().IsSameSize(input.shape()), in Compute()
103 first_input.shape().DebugString(), " != values[", i, in Compute()
|
| D | concat_op.cc | 99 const Tensor& first_input = c->input(values_input_start_index_); in Compute() local
100 const int input_dims = first_input.dims(); in Compute()
101 const TensorShape& input_shape = first_input.shape(); in Compute()
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | control_flow_ops.cc | 101 ShapeHandle first_input = c->input(1); in __anon92ff84320202() local
102 if (!c->FullyDefined(first_input)) { in __anon92ff84320202()
110 !c->Merge(first_input, input, &unused).ok()) { in __anon92ff84320202()
115 c->set_output(0, first_input); in __anon92ff84320202()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | executor.cc | 295 const TaggedNode& tagged_node, Entry* first_input,
302 Status PrepareInputs(const NodeItem& item, Entry* first_input,
500 first_input(_first_input), in AsyncState()
514 Entry* first_input; member
594 const TaggedNode& tagged_node, Entry* first_input, in ProcessAsync() argument
599 new AsyncState(params, tagged_node, &item, first_input, stats); in ProcessAsync()
604 Entry* first_input = state->first_input; // Shorthand in ProcessAsync() local
620 (first_input + i)->ClearVal(); in ProcessAsync()
770 Entry* first_input = propagator_.GetInputTensors(tagged_node); in Process() local
785 s = PrepareInputs(item, first_input, &inputs, &input_alloc_attrs, in Process()
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | single_threaded_executor.cc | 334 Entry& first_input = inputs[arg_output_locations_[i][0]]; in Run() local
335 first_input.state = Entry::State::HAS_VALUE; in Run()
336 first_input.val.Init(); in Run()
337 args.call_frame->ConsumeArg(i, first_input.val.get()); in Run()
347 input.val.Init(*first_input.val); in Run()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | rnn.py | 1326 first_input = inputs
1327 while nest.is_sequence(first_input):
1328 first_input = first_input[0]
1332 if first_input.get_shape().rank != 1:
1334 input_shape = first_input.get_shape().with_rank_at_least(2)
1349 fixed_batch_size = first_input.get_shape().with_rank_at_least(1)[0]
1354 batch_size = array_ops.shape(first_input)[0]
|
| /external/mesa3d/src/gallium/auxiliary/vl/ |
| D | vl_mc.h | 75 unsigned first_input,
|
| D | vl_idct.h | 88 unsigned first_input, struct ureg_dst fragment);
|
| D | vl_idct.c | 432 unsigned first_input, struct ureg_dst fragment) in vl_idct_stage2_frag_shader() argument
438 --first_input; in vl_idct_stage2_frag_shader()
440 …l_addr[0] = ureg_DECL_fs_input(shader, TGSI_SEMANTIC_GENERIC, first_input + VS_O_L_ADDR0, TGSI_INT… in vl_idct_stage2_frag_shader()
441 …l_addr[1] = ureg_DECL_fs_input(shader, TGSI_SEMANTIC_GENERIC, first_input + VS_O_L_ADDR1, TGSI_INT… in vl_idct_stage2_frag_shader()
443 …r_addr[0] = ureg_DECL_fs_input(shader, TGSI_SEMANTIC_GENERIC, first_input + VS_O_R_ADDR0, TGSI_INT… in vl_idct_stage2_frag_shader()
444 …r_addr[1] = ureg_DECL_fs_input(shader, TGSI_SEMANTIC_GENERIC, first_input + VS_O_R_ADDR1, TGSI_INT… in vl_idct_stage2_frag_shader()
|
| D | vl_mpeg12_decoder.c | 1089 unsigned first_input, in mc_frag_shader_callback() argument
1100 vl_idct_stage2_frag_shader(idct, shader, first_input, dst); in mc_frag_shader_callback()
1102 src = ureg_DECL_fs_input(shader, TGSI_SEMANTIC_GENERIC, first_input, TGSI_INTERPOLATE_LINEAR); in mc_frag_shader_callback()
|
| /external/tensorflow/tensorflow/python/compiler/tensorrt/ |
| D | trt_convert.py | 1131 first_input = None
1136 if not first_input:
1137 first_input = inp
1149 self._converted_func(*map(ops.convert_to_tensor, first_input))
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ir/ |
| D | tfl_ops.cc | 1498 Operation *first_input = pack_op.getOperand(0).getDefiningOp(); in matchAndRewrite() local
1499 if (!first_input) return failure(); in matchAndRewrite()
1500 auto input_unpack_op = dyn_cast_or_null<TFL::UnpackOp>(first_input); in matchAndRewrite()
|
| /external/swiftshader/third_party/SPIRV-Tools/source/opt/ |
| D | folding_rules.cpp | 1654 Instruction* first_input = in VectorShuffleFeedingExtract() local
1657 type_mgr->GetType(first_input->type_id()); in VectorShuffleFeedingExtract()
|
| /external/deqp-deps/SPIRV-Tools/source/opt/ |
| D | folding_rules.cpp | 1654 Instruction* first_input = in VectorShuffleFeedingExtract() local
1657 type_mgr->GetType(first_input->type_id()); in VectorShuffleFeedingExtract()
|
| /external/angle/third_party/vulkan-deps/spirv-tools/src/source/opt/ |
| D | folding_rules.cpp | 1707 Instruction* first_input = in VectorShuffleFeedingExtract() local
1710 type_mgr->GetType(first_input->type_id()); in VectorShuffleFeedingExtract()
|
| /external/tensorflow/tensorflow/python/ops/parallel_for/ |
| D | pfor.py | 3184 first_input, first_input_stacked, _ = pfor_input.input(0)
3213 result = xla.einsum(equation=new_equation, a=first_input, b=second_input)
3216 result = special_math_ops.einsum(new_equation, first_input, second_input)
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | arithmetic_optimizer.cc | 1706 const int first_input = concat_node->op() == "Concat" ? 1 : 0; in HoistChainForConcat() local
1718 if (concat_port == first_input) { in HoistChainForConcat()
|