Searched defs:params_shape (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/core/ops/ |
| D | cudnn_rnn_ops.cc | 228 auto params_shape = c->input(3); in __anon815c5c9f0602() local 265 auto params_shape = c->input(3); in __anon815c5c9f0702() local 305 auto params_shape = c->input(3); in __anon815c5c9f0802() local
|
| D | array_ops.cc | 1186 ShapeHandle params_shape; in __anon42d741191802() local
|
| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | ragged_gather_op_test.py | 355 params_shape, argument
|
| /external/tensorflow/tensorflow/python/kernel_tests/array_ops/ |
| D | gather_op_test.py | 538 def testBatchDimsMatchesPythonBatching(self, params_shape, indices_shape, argument
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | gather_op.cc | 275 TensorShape params_shape = context->InputShape(0); in Compile() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | scatter_nd_op.cc | 278 TensorShape params_shape; in DoCompute() local 777 Status ValidateUpdateShape(const TensorShape& params_shape, in ValidateUpdateShape() 818 Status PrepareAndValidateInputs(const TensorShape& params_shape, in PrepareAndValidateInputs()
|
| D | cudnn_rnn_ops.cc | 2002 const TensorShape& params_shape, in AllocateOutputs()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | array_grad.py | 585 def _GetBatchIndices(params_shape, indices, batch_dims): argument 606 def _BatchGatherGrad(params_shape, values, indices, batch_dims, argument
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | reference_ops.h | 1068 inline GatherNdHelperResult GatherNdHelper(const RuntimeShape& params_shape, in GatherNdHelper() 1094 inline void GatherNd(const RuntimeShape& params_shape, in GatherNd() 1114 inline void GatherNdString(const RuntimeShape& params_shape, in GatherNdString()
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | resource_variable_ops_test.py | 1528 def testGatherWithBatchDimsMatchesTensor(self, params_shape, indices_shape, argument
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes_test.cc | 5264 std::vector<int> params_shape; in TEST_P() member
|