| /external/tensorflow/tensorflow/core/ops/ |
| D | image_ops_test.cc | 48 Tensor size_tensor = test::AsTensor<int32>({20, 30}); in TEST() local
233 Tensor size_tensor = test::AsTensor<int32>({20, 30}); in TEST() local
261 Tensor size_tensor = test::AsTensor<int32>({20, 30}); in TEST() local
287 Tensor size_tensor = test::AsTensor<int32>({20, 30}); in TEST() local
346 const Tensor size_tensor = test::AsTensor<int32>({20, 30}); in TEST() local
|
| D | math_ops.cc | 1710 const Tensor* size_tensor = c->input_tensor(1); in __anon328d24420e02() local
1747 const Tensor* size_tensor = c->input_tensor(1); in __anon328d24420f02() local
1792 const Tensor* size_tensor = c->input_tensor(3); in __anon328d24421002() local
|
| D | image_ops.cc | 39 const Tensor* size_tensor = c->input_tensor(size_input_idx); in SetOutputToSizedImage() local
|
| /external/tensorflow/tensorflow/lite/delegates/hexagon/builders/ |
| D | resize_bilinear_builder.cc | 35 const auto& size_tensor = context->tensors[inputs->data[1]]; in PopulateSubGraph() local
|
| D | slice_builder.cc | 45 const auto& size_tensor = context->tensors[inputs->data[2]]; in PopulateSubGraph() local
|
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | backports_test.cc | 114 Tensor size_tensor(DT_INT32, {}); in TEST() local
|
| D | quantize_nodes_test.cc | 659 Tensor size_tensor(DT_INT32, TensorShape({2})); in TestQuantizeResizeBilinear() local
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ops/ |
| D | slice_ops.cc | 229 nvinfer1::ITensor* size_tensor; in HandleDynamicStridedSliceInput() local
|
| D | einsum.cc | 301 StatusOr<nvinfer1::IConstantLayer*> size_tensor = in SetDynamicSize() local
|
| /external/tensorflow/tensorflow/lite/delegates/hexagon/ |
| D | utils.cc | 384 const auto& size_tensor = context->tensors[node->inputs->data[1]]; in IsNodeSupportedByHexagon() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | slice_op.cc | 69 const Tensor& size_tensor = context->input(2); in SharedSliceValidation() local
|
| D | bincount_op.cc | 212 const Tensor& size_tensor = ctx->input(1); in Compute() local
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | hashtable_ops_test.cc | 276 auto* size_tensor = interpreter_->tensor(kSizeTensorId); in GetTableSize() local
282 auto* size_tensor = interpreter_->tensor(kSizeTwoTensorId); in GetTableTwoSize() local
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_slice_op.cc | 74 const Tensor& size_tensor = MklGetInput(context, kInputSizeIndex); in ValidateMklInputs() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | tensor_array_ops.cc | 561 Tensor size_tensor(DT_INT32, {}); in Compile() local
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes.cc | 2590 Tensor size_tensor = tensor::DeepCopy(size_weights.GetTensor()); in ConvertSlice() local
|