Searched refs:params_tensor (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | scatter_op_test.cc | 58 Tensor params_tensor = *mutable_input(0).tensor; in TEST_F() local
61 test::ExpectTensorEqual<string>(expected, params_tensor); in TEST_F()
71 Tensor params_tensor = *mutable_input(0).tensor; in TEST_F() local
74 test::ExpectTensorEqual<bool>(expected, params_tensor); in TEST_F()
89 Tensor params_tensor = *mutable_input(0).tensor; in TEST_F() local
93 test::ExpectTensorEqual<float>(expected, params_tensor); in TEST_F()
108 Tensor params_tensor = *mutable_input(0).tensor; in TEST_F() local
112 test::ExpectTensorEqual<float>(expected, params_tensor); in TEST_F()
125 Tensor params_tensor = *mutable_input(0).tensor; in TEST_F() local
128 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 | 77 Tensor* params_tensor = GetOutput(0); in Reverse_234() local
86 test::ExpectTensorEqual<T>(expected, *params_tensor); in Reverse_234()
103 Tensor* params_tensor = GetOutput(0); in Reverse_1234() local
112 test::ExpectTensorEqual<T>(expected, *params_tensor); in Reverse_1234()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes.cc | 3728 TRT_TensorOrWeights params_tensor = inputs.at(0); in ConvertGather() local
3736 const int tf_gather_output_rank = params_tensor.GetTrtDims().nbDims + in ConvertGather()
3751 *const_cast<nvinfer1::ITensor*>(params_tensor.tensor()), in ConvertGather()
|