Home
last modified time | relevance | path

Searched defs:inputs_shape (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dctc_decoder_ops.cc68 const TensorShape& inputs_shape = (*inputs)->shape(); in ValidateInputsGenerateOutputs() local
204 const TensorShape& inputs_shape = inputs->shape(); in Compute() local
313 const TensorShape& inputs_shape = inputs->shape(); in Compute() local
H A Dctc_loss_op.cc112 const TensorShape& inputs_shape = inputs->shape(); in Compute() local
283 const TensorShape& inputs_shape = inputs->shape(); in Compute() local
/aosp_15_r20/external/tensorflow/tensorflow/python/keras/layers/legacy_rnn/
H A Drnn_cell_impl.py458 def build(self, inputs_shape): argument
569 def build(self, inputs_shape): argument
749 def build(self, inputs_shape): argument
973 def build(self, inputs_shape): argument
H A Drnn_cell_wrapper_impl.py193 def build(self, inputs_shape): argument
/aosp_15_r20/external/tensorflow/tensorflow/python/keras/layers/
H A Drnn_cell_wrapper_v2.py70 def build(self, inputs_shape): argument
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/
H A Darithmetic_optimizer_test.cc1265 Output inputs_shape = ops::Shape(s, inputs); in TEST_F() local
1307 Output inputs_shape = ops::Shape(s, inputs); in TEST_F() local
1801 Output inputs_shape = in TEST_F() local
1865 Output inputs_shape = in TEST_F() local
1943 Output inputs_shape = in TEST_F() local
1967 Output inputs_shape = in TEST_F() local