Searched refs:input_elements (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | list_dataset_op_test.cc | 76 const std::vector<std::vector<Tensor>>& input_elements) { in ListInputTypes() argument
78 for (const auto& element : input_elements) { in ListInputTypes()
87 const std::vector<std::vector<Tensor>>& input_elements) { in ListOutputTypes() argument
89 for (const auto& tensor : input_elements.front()) { in ListOutputTypes()
96 const std::vector<std::vector<Tensor>>& input_elements) { in ListOutputShapes() argument
98 for (const auto& tensor : input_elements.front()) { in ListOutputShapes()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | while_loop_concat_code_motion.cc | 132 HloInstruction* CreateConcat(std::vector<HloInstruction*> input_elements, in CreateConcat()
135 for (int64_t i = 0; i < input_elements.size(); ++i) { in CreateConcat()
137 element_shape.reserve(input_elements[i]->shape().rank() + 1); in CreateConcat()
138 for (int64_t j = 0; j < input_elements[i]->shape().rank(); ++j) { in CreateConcat()
142 element_shape.push_back(input_elements[i]->shape().dimensions(j)); in CreateConcat()
147 input_elements[i] = comp->AddInstruction(HloInstruction::CreateReshape( in CreateConcat()
148 ShapeUtil::MakeShape(input_elements[i]->shape().element_type(), in CreateConcat()
150 input_elements[i])); in CreateConcat()
155 GetConcatShape(), input_elements, concat_dim)); in CreateConcat()
|
| /external/XNNPACK/src/operators/ |
| D | average-pooling-nhwc.c | 693 const size_t input_elements = input_height * input_width; in setup_average_pooling2d() local
700 .input_batch_stride = input_stride_in_bytes * input_elements, in setup_average_pooling2d()
701 .input_elements = input_elements, in setup_average_pooling2d()
710 if (input_elements <= gavgpool->row_tile) { in setup_average_pooling2d()
|
| D | global-average-pooling-ncw.c | 130 .input_elements = width * sizeof(float), in xnn_setup_global_average_pooling_ncw_f32()
|
| D | global-average-pooling-nwc.c | 180 .input_elements = width, in setup_global_average_pooling_nwc()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | shape_inference.cc | 1086 DatasetInput input_elements = GetDatasetInput(op->getOperand(0)); in InferShapeForDatasetOpCommon() local
1087 if (!input_elements) { in InferShapeForDatasetOpCommon()
1097 Type t = GetType(input_elements.shapes[i], input_elements.types[i]); in InferShapeForDatasetOpCommon()
1164 DatasetInput input_elements = GetDatasetInput(op.input_dataset()); in InferShapeForReduceDataset() local
1174 if (input_elements) { in InferShapeForReduceDataset()
1175 num_input_elements = input_elements.shapes.size(); in InferShapeForReduceDataset()
1209 if (input_elements) { in InferShapeForReduceDataset()
1210 Type t = GetType(input_elements.shapes[i], input_elements.types[i]); in InferShapeForReduceDataset()
|
| /external/pytorch/aten/src/ATen/native/quantized/cpu/qnnpack/src/ |
| D | operator-run.c | 646 size_t input_elements; member
667 context->input_elements, in compute_global_average_pooling_unipass()
692 context->input_elements, in compute_global_average_pooling_multipass()
1500 .input_elements = input_width, in pytorch_qnnp_run_operator()
|
| /external/XNNPACK/src/xnnpack/ |
| D | compute.h | 713 size_t input_elements; member
740 size_t input_elements; member
|
| /external/XNNPACK/src/ |
| D | operator-run.c | 845 context->input_elements, in xnn_compute_global_average_pooling_nwc_unipass()
867 context->input_elements, in xnn_compute_global_average_pooling_nwc_multipass()
889 context->input_elements, in xnn_compute_global_average_pooling_ncw()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | fully_connected.cc | 925 const int input_elements = input_shape.FlatSize(); in VerifySparsity() local
938 if (input_elements <= in VerifySparsity()
|