Home
last modified time | relevance | path

Searched defs:inputs_size (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/lite/python/optimize/
Dcalibration_wrapper.cc250 const size_t inputs_size = PyList_Size(input_shapes); in Prepare() local
282 const size_t inputs_size = PyList_Size(input_shapes); in Prepare() local
321 const size_t inputs_size = PyList_Size(input_value); in FeedTensor() local
354 const size_t inputs_size = PyList_Size(input_value); in FeedTensor() local
/external/tensorflow/tensorflow/python/ops/parallel_for/
Dgradients_test.py68 def lstm_model_fn(batch_size, state_size, steps, inputs_size=None): argument
107 def create_lstm_batch_jacobian(batch_size, state_size, steps, inputs_size=None): argument
181 def create_lstm_per_eg_grad(batch_size, state_size, steps, inputs_size=None): argument
/external/tensorflow/tensorflow/core/kernels/data/
Dflat_map_dataset_op.cc267 size_t inputs_size; in RestoreInternal() local
Dparallel_interleave_dataset_op.cc1388 int64_t inputs_size; in ReadElement() local
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf_control_flow.cc361 const int inputs_size = inputs.size(); in LowerWhileRegion() local
/external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/
Drecording_tpu_driver.cc425 const auto inputs_size = inputs.size(); in ExecuteProgram() local
/external/tensorflow/tensorflow/core/framework/
Dshape_inference.cc221 const int inputs_size = inputs_.size(); in PostInputInit() local