| /aosp_15_r20/external/tensorflow/tensorflow/core/ops/ |
| H A 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
|
| H A D | math_ops.cc | 1710 const Tensor* size_tensor = c->input_tensor(1); in __anon247ac1530e02() local
1747 const Tensor* size_tensor = c->input_tensor(1); in __anon247ac1530f02() local
1792 const Tensor* size_tensor = c->input_tensor(3); in __anon247ac1531002() local
|
| H A D | image_ops.cc | 39 const Tensor* size_tensor = c->input_tensor(size_input_idx); in SetOutputToSizedImage() local
|
| /aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/optim/ |
| H A D | serialize.h | 96 torch::Tensor size_tensor; in serialize() local
232 torch::Tensor size_tensor; in serialize() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/ |
| H A D | resize_bilinear_builder.cc | 35 const auto& size_tensor = context->tensors[inputs->data[1]]; in PopulateSubGraph() local
|
| H A D | slice_builder.cc | 45 const auto& size_tensor = context->tensors[inputs->data[2]]; in PopulateSubGraph() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/tools/graph_transforms/ |
| H A D | backports_test.cc | 114 Tensor size_tensor(DT_INT32, {}); in TEST() local
|
| H A D | quantize_nodes_test.cc | 659 Tensor size_tensor(DT_INT32, TensorShape({2})); in TestQuantizeResizeBilinear() local
|
| /aosp_15_r20/external/federated-compute/fcp/tensorflow/ |
| H A D | dictionary_ops.cc | 109 tf::Tensor* size_tensor; in DoCompute() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ops/ |
| H A D | slice_ops.cc | 229 nvinfer1::ITensor* size_tensor; in HandleDynamicStridedSliceInput() local
|
| H A D | einsum.cc | 301 StatusOr<nvinfer1::IConstantLayer*> size_tensor = in SetDynamicSize() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/ |
| H A 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
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/hexagon/ |
| H A D | utils.cc | 384 const auto& size_tensor = context->tensors[node->inputs->data[1]]; in IsNodeSupportedByHexagon() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | slice_op.cc | 69 const Tensor& size_tensor = context->input(2); in SharedSliceValidation() local
|
| H A D | bincount_op.cc | 212 const Tensor& size_tensor = ctx->input(1); in Compute() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/ |
| H A D | mkl_slice_op.cc | 74 const Tensor& size_tensor = MklGetInput(context, kInputSizeIndex); in ValidateMklInputs() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| H A D | tensor_array_ops.cc | 561 Tensor size_tensor(DT_INT32, {}); in Compile() local
|
| /aosp_15_r20/external/pytorch/test/cpp/jit/ |
| H A D | test_custom_class_registrations.cpp | 136 at::Tensor size_tensor; in TensorQueue() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| H A D | convert_nodes.cc | 2590 Tensor size_tensor = tensor::DeepCopy(size_weights.GetTensor()); in ConvertSlice() local
|