Searched refs:params_tensor (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | scatter_op_test.cc | 69 Tensor params_tensor = *mutable_input(0).tensor; in TEST_F() local
72 test::ExpectTensorEqual<tstring>(expected, params_tensor); in TEST_F()
82 Tensor params_tensor = *mutable_input(0).tensor; in TEST_F() local
85 test::ExpectTensorEqual<bool>(expected, params_tensor); in TEST_F()
100 Tensor params_tensor = *mutable_input(0).tensor; in TEST_F() local
104 test::ExpectTensorEqual<float>(expected, params_tensor); in TEST_F()
119 Tensor params_tensor = *mutable_input(0).tensor; in TEST_F() local
123 test::ExpectTensorEqual<float>(expected, params_tensor); in TEST_F()
136 Tensor params_tensor = *mutable_input(0).tensor; in TEST_F() local
139 test::ExpectTensorEqual<float>(expected, params_tensor); in TEST_F()
[all …]
|
| D | scatter_nd_op_test.cc | 90 Tensor params_tensor = *mutable_input(0).tensor; in TEST_F() local
94 test::ExpectTensorEqual<float>(expected, params_tensor); in TEST_F()
109 Tensor params_tensor = *mutable_input(0).tensor; in TEST_F() local
113 test::ExpectTensorEqual<float>(expected, params_tensor); in TEST_F()
139 Tensor params_tensor = *mutable_input(0).tensor; in TEST_F() local
142 test::ExpectTensorEqual<float>(expected, params_tensor); in TEST_F()
155 Tensor params_tensor = *mutable_input(0).tensor; in TEST_F() local
158 test::ExpectTensorEqual<float>(expected, params_tensor); in TEST_F()
171 Tensor params_tensor = *mutable_input(0).tensor; in TEST_F() local
174 test::ExpectTensorEqual<float>(expected, params_tensor); in TEST_F()
|
| D | reverse_op_test.cc | 79 Tensor* params_tensor = GetOutput(0); in Reverse_234() local
88 test::ExpectTensorEqual<T>(expected, *params_tensor); in Reverse_234()
105 Tensor* params_tensor = GetOutput(0); in Reverse_1234() local
114 test::ExpectTensorEqual<T>(expected, *params_tensor); in Reverse_1234()
|
| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | ragged_gather_op_test.py | 382 for params_tensor in params_tensors:
385 params_tensor, indices_tensor, axis=axis, batch_dims=batch_dims)
390 (getattr(params_tensor, 'ragged_rank',
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes.cc | 5366 nvinfer1::ITensor* params_tensor = nullptr; in ConvertGather() local
5368 params_tensor = params->converter->CreateConstantLayer( in ConvertGather()
5371 params_tensor = params_input.tensor(); in ConvertGather()
5380 *params_tensor, *indices_input.tensor(), trt_axis); in ConvertGather()
|