Home
last modified time | relevance | path

Searched defs:params_shape (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dscatter_nd_op.cc130 TensorShape params_shape; in DoCompute() local
294 Status ValidateUpdateShape(const TensorShape& params_shape, in ValidateUpdateShape()
330 Status PrepareAndValidateInputs(const TensorShape& params_shape, in PrepareAndValidateInputs()
Dgather_nd_op.cc123 const TensorShape& params_shape(params.shape()); in DoGatherNd() local
/external/tensorflow/tensorflow/contrib/cudnn_rnn/ops/
Dcudnn_rnn_ops.cc204 auto params_shape = c->input(3); in __anone2c774540402() local
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dgather_op.cc221 TensorShape params_shape = context->InputShape(0); in Compile() local
/external/tensorflow/tensorflow/core/ops/
Darray_ops.cc911 ShapeHandle params_shape; in __anondb9326b21602() local