Searched defs:params_shape (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/core/ops/ |
| D | cudnn_rnn_ops.cc | 225 auto params_shape = c->input(3); in __anone1d60d010602() local
262 auto params_shape = c->input(3); in __anone1d60d010702() local
301 auto params_shape = c->input(3); in __anone1d60d010802() local
|
| D | array_ops.cc | 1118 ShapeHandle params_shape; in __anon7c94107b1802() local
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | gather_op_test.py | 407 def testBatchDimsMatchesPythonBatching(self, params_shape, indices_shape, argument
|
| D | resource_variable_ops_test.py | 1205 def testGatherWithBatchDimsMatchesTensor(self, params_shape, indices_shape, argument
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | gather_op.cc | 206 TensorShape params_shape = context->InputShape(0); in Compile() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | scatter_nd_op.cc | 271 TensorShape params_shape; in DoCompute() local
518 Status ValidateUpdateShape(const TensorShape& params_shape, in ValidateUpdateShape()
554 Status PrepareAndValidateInputs(const TensorShape& params_shape, in PrepareAndValidateInputs()
|
| D | gather_nd_op.cc | 123 const TensorShape& params_shape(params.shape()); in DoGatherNd() local
|
| D | cudnn_rnn_ops.cc | 1802 const TensorShape& params_shape, in AllocateOutputs()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | reference_ops.h | 2973 inline void GatherNd(const RuntimeShape& params_shape, in GatherNd()
|