| /external/tensorflow/tensorflow/core/ops/ |
| D | image_ops_test.cc | 48 Tensor size_tensor = test::AsTensor<int32>({20, 30}); in TEST() local
49 op.input_tensors[1] = &size_tensor; in TEST()
233 Tensor size_tensor = test::AsTensor<int32>({20, 30}); in TEST() local
234 op.input_tensors[1] = &size_tensor; in TEST()
261 Tensor size_tensor = test::AsTensor<int32>({20, 30}); in TEST() local
262 op.input_tensors[3] = &size_tensor; in TEST()
287 Tensor size_tensor = test::AsTensor<int32>({20, 30}); in TEST() local
288 op.input_tensors[1] = &size_tensor; in TEST()
346 const Tensor size_tensor = test::AsTensor<int32>({20, 30}); in TEST() local
347 op.input_tensors.at(1) = &size_tensor; in TEST()
|
| D | math_ops.cc | 1710 const Tensor* size_tensor = c->input_tensor(1); in __anonfffccbd30e02() local
1711 if (size_tensor == nullptr) { in __anonfffccbd30e02()
1717 if (size_tensor->dims() != 0) { in __anonfffccbd30e02()
1719 size_tensor->dims()); in __anonfffccbd30e02()
1723 int32_t size_val = size_tensor->scalar<int32>()(); in __anonfffccbd30e02()
1747 const Tensor* size_tensor = c->input_tensor(1); in __anonfffccbd30f02() local
1748 if (size_tensor == nullptr) { in __anonfffccbd30f02()
1753 if (size_tensor->dims() != 0) { in __anonfffccbd30f02()
1755 size_tensor->dims()); in __anonfffccbd30f02()
1762 size_val = static_cast<int64_t>(size_tensor->scalar<int32>()()); in __anonfffccbd30f02()
[all …]
|
| D | image_ops.cc | 39 const Tensor* size_tensor = c->input_tensor(size_input_idx); in SetOutputToSizedImage() local
42 if (size_tensor == nullptr) { in SetOutputToSizedImage()
47 if (size_tensor->dtype() != DT_INT32) { in SetOutputToSizedImage()
51 DataTypeString(size_tensor->dtype()), " for input #", size_input_idx, in SetOutputToSizedImage()
54 auto vec = size_tensor->vec<int32>(); in SetOutputToSizedImage()
|
| /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
36 if (!IsConstantTensor(&size_tensor)) { in PopulateSubGraph()
43 std::vector<int> dims = {size_tensor.data.i32[0], size_tensor.data.i32[1]}; in PopulateSubGraph()
|
| D | slice_builder.cc | 45 const auto& size_tensor = context->tensors[inputs->data[2]]; in PopulateSubGraph() local
49 &size_tensor, &begins, &sizes); in PopulateSubGraph()
52 &size_tensor, &begins, &sizes); in PopulateSubGraph()
|
| /external/pytorch/torch/csrc/api/include/torch/optim/ |
| D | serialize.h | 96 torch::Tensor size_tensor; in serialize() local
97 param_group_archive.read("params/size", size_tensor); in serialize()
98 const int64_t size = size_tensor.item<int64_t>(); in serialize()
232 torch::Tensor size_tensor; in serialize() local
233 archive.read(key + "/size", size_tensor); in serialize()
234 const size_t size = size_tensor.item<int64_t>(); in serialize()
|
| /external/executorch/backends/transforms/ |
| D | mean_to_sum_div.py | 27 size_tensor = super().call_operator(
40 exir_ops.edge.aten.div.Tensor, (sum_res, size_tensor), {}, meta
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | slice_op.cc | 69 const Tensor& size_tensor = context->input(2); in SharedSliceValidation() local
74 TensorShapeUtils::IsVector(size_tensor.shape()) && in SharedSliceValidation()
76 size_tensor.NumElements() == input.dims(), in SharedSliceValidation()
80 " and ", size_tensor.shape().DebugString(), " instead.")); in SharedSliceValidation()
84 IntTensorToInt64Vec(size_tensor, size); in SharedSliceValidation()
|
| D | bincount_op.cc | 212 const Tensor& size_tensor = ctx->input(1); in Compute() local
213 OP_REQUIRES(ctx, size_tensor.dims() == 0, in Compute()
215 size_tensor.dims())); in Compute()
216 int32_t size = size_tensor.scalar<int32_t>()(); in Compute()
|
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | backports_test.cc | 114 Tensor size_tensor(DT_INT32, {}); in TEST() local
115 size_tensor.flat<int32>()(0) = 1; in TEST()
116 SetNodeTensorAttr<float>("value", size_tensor, size_node); in TEST()
|
| D | quantize_nodes_test.cc | 659 Tensor size_tensor(DT_INT32, TensorShape({2})); in TestQuantizeResizeBilinear() local
660 test::FillValues<int32>(&size_tensor, {256, 256}); in TestQuantizeResizeBilinear()
663 Input::Initializer(size_tensor)); in TestQuantizeResizeBilinear()
|
| /external/federated-compute/fcp/tensorflow/ |
| D | dictionary_ops.cc | 109 tf::Tensor* size_tensor; in DoCompute() local
111 context->allocate_output(0, tf::TensorShape({}), &size_tensor); in DoCompute()
119 size_tensor->flat<int64_t>()(0) = dictionary.Size(); in DoCompute()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ops/ |
| D | slice_ops.cc | 229 nvinfer1::ITensor* size_tensor; in HandleDynamicStridedSliceInput() local
237 size_tensor = (*ceil_div)->getOutput(0); in HandleDynamicStridedSliceInput()
241 slice_layer->setInput(2, *size_tensor); in HandleDynamicStridedSliceInput()
|
| D | einsum.cc | 301 StatusOr<nvinfer1::IConstantLayer*> size_tensor = in SetDynamicSize() local
303 TRT_ENSURE_PTR_OK(size_tensor); in SetDynamicSize()
304 size_tensors[i] = (*size_tensor)->getOutput(0); in SetDynamicSize()
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_slice_op.cc | 74 const Tensor& size_tensor = MklGetInput(context, kInputSizeIndex); in ValidateMklInputs() local
93 TensorShapeUtils::IsVector(size_tensor.shape()) && in ValidateMklInputs()
95 size_tensor.NumElements() == input_dims, in ValidateMklInputs()
99 " and ", size_tensor.shape().DebugString(), " instead.")); in ValidateMklInputs()
102 *size = IntTensorToInt64Vec(size_tensor); in ValidateMklInputs()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | hashtable_ops_test.cc | 276 auto* size_tensor = interpreter_->tensor(kSizeTensorId); in GetTableSize() local
277 auto size_tensor_shape = GetTensorShape(size_tensor); in GetTableSize()
278 return GetTensorData<int64_t>(size_tensor)[0]; in GetTableSize()
282 auto* size_tensor = interpreter_->tensor(kSizeTwoTensorId); in GetTableTwoSize() local
283 auto size_tensor_shape = GetTensorShape(size_tensor); in GetTableTwoSize()
284 return GetTensorData<int64_t>(size_tensor)[0]; in GetTableTwoSize()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | tensor_array_ops.cc | 561 Tensor size_tensor(DT_INT32, {}); in Compile() local
562 size_tensor.scalar<int32>()() = static_cast<int32>(var->max_array_size()); in Compile()
563 ctx->SetConstantOutput(0, size_tensor); in Compile()
|
| /external/tensorflow/tensorflow/lite/delegates/hexagon/ |
| D | utils.cc | 384 const auto& size_tensor = context->tensors[node->inputs->data[1]]; in IsNodeSupportedByHexagon() local
385 return NumElements(&size_tensor) == 2; in IsNodeSupportedByHexagon()
|
| /external/pytorch/test/cpp/jit/ |
| D | test_custom_class_registrations.cpp | 136 at::Tensor size_tensor; in TensorQueue() local
137 size_tensor = dict.at(std::string(key + "/size")).cpu(); in TensorQueue()
138 const auto* size_tensor_acc = size_tensor.const_data_ptr<int64_t>(); in TensorQueue()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes.cc | 2590 Tensor size_tensor = tensor::DeepCopy(size_weights.GetTensor()); in ConvertSlice() local
2596 auto size_vec = size_tensor.flat<int32>(); in ConvertSlice()
|