Searched refs:params_dims (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | gather_op.cc | 169 const auto params_dims = input_shape.dims(); in Compile() local
171 context, -params_dims <= axis && axis < params_dims, in Compile()
172 errors::InvalidArgument("Expected axis in the range [", -params_dims, in Compile()
173 ", ", params_dims, "), but got ", axis)); in Compile()
175 axis += params_dims; in Compile()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes_test.cc | 3437 std::vector<int> params_dims; in TestConvertGather() member
3469 test->AddTestTensor("params", ok_params[i].params_dims, 1, in TestConvertGather()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | reference_ops.h | 2984 const int params_dims = params_shape.DimensionsCount(); in GatherNd() local
2988 for (int i = indices_nd; i < params_dims; ++i) { in GatherNd()
|