Searched defs:inputs_shape (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/lite/experimental/examples/lstm/ |
| D | rnn_cell.py | 98 def build(self, inputs_shape): argument
296 def build(self, inputs_shape): argument
|
| /external/tensorflow/tensorflow/lite/micro/kernels/ |
| D | concatenation.cc | 89 RuntimeShape inputs_shape[kMaxInputNum]; in EvalUnquantized() local
109 RuntimeShape inputs_shape[kMaxInputNum]; in EvalQuantizedUInt8() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | ctc_decoder_ops.cc | 68 const TensorShape& inputs_shape = (*inputs)->shape(); in ValidateInputsGenerateOutputs() local
200 const TensorShape& inputs_shape = inputs->shape(); in Compute() local
302 const TensorShape& inputs_shape = inputs->shape(); in Compute() local
|
| D | ctc_loss_op.cc | 106 const TensorShape& inputs_shape = inputs->shape(); in Compute() local
274 const TensorShape& inputs_shape = inputs->shape(); in Compute() local
|
| /external/tensorflow/tensorflow/python/keras/layers/legacy_rnn/ |
| D | rnn_cell_impl.py | 458 def build(self, inputs_shape): argument
568 def build(self, inputs_shape): argument
746 def build(self, inputs_shape): argument
969 def build(self, inputs_shape): argument
|
| D | rnn_cell_wrapper_impl.py | 197 def build(self, inputs_shape): argument
|
| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | rnn_cell_wrapper_v2.py | 74 def build(self, inputs_shape): argument
|
| D | recurrent_test.py | 1859 def build(self, inputs_shape): argument
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | arithmetic_optimizer_test.cc | 837 Output inputs_shape = ops::Shape(s, inputs); in TEST_F() local
879 Output inputs_shape = ops::Shape(s, inputs); in TEST_F() local
1342 Output inputs_shape = in TEST_F() local
1406 Output inputs_shape = in TEST_F() local
1484 Output inputs_shape = in TEST_F() local
1508 Output inputs_shape = in TEST_F() local
|