Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/lite/python/optimize/
H A 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
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/parallel_for/
H A 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
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/
H A Dflat_map_dataset_op.cc267 size_t inputs_size; in RestoreInternal() local
H A Dparallel_interleave_dataset_op.cc1388 int64_t inputs_size; in ReadElement() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
H A Dlegalize_tf_control_flow.cc361 const int inputs_size = inputs.size(); in LowerWhileRegion() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/
H A Drecording_tpu_driver.cc425 const auto inputs_size = inputs.size(); in ExecuteProgram() local
/aosp_15_r20/external/executorch/runtime/executor/
H A Dmethod.cpp1333 size_t Method::inputs_size() const { in inputs_size() function in executorch::runtime::Method
/aosp_15_r20/external/executorch/extension/pybindings/
H A Dpybindings.cpp672 const auto inputs_size = py::len(inputs); in run_method() local
/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A Dshape_inference.cc221 const int inputs_size = inputs_.size(); in PostInputInit() local
/aosp_15_r20/out/soong/.intermediates/external/tensorflow/tensorflow_core_proto_cpp_lite/android_arm64_armv8-2a_cortex-a55_static_apex33/gen/proto/tensorflow/core/framework/
Dmodel.pb.h1514 inline int ModelProto_Node::inputs_size() const { in inputs_size() function
/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/quantization/
H A Dinsert_quant_dequant.cpp358 const auto inputs_size = embedding_bag_float_op->inputs().size(); in insertEmbeddingBagOps() local
/aosp_15_r20/out/soong/.intermediates/external/tensorflow/tensorflow_core_proto_cpp_lite/android_arm64_armv8-2a_cortex-a55_static_apex33/gen/proto/tensorflow/core/grappler/costs/
Dop_performance_data.pb.h1918 inline int OpInfo::inputs_size() const { in inputs_size() function
/aosp_15_r20/out/soong/.intermediates/external/tensorflow/tensorflow_core_proto_cpp_lite/android_arm64_armv8-2a_cortex-a55_static_apex33/gen/proto/tensorflow/core/protobuf/
Dmeta_graph.pb.h4866 inline int SignatureDef::inputs_size() const { in inputs_size() function
/aosp_15_r20/out/soong/.intermediates/frameworks/base/tools/aapt2/libaapt2/linux_glibc_x86_64_static/gen/proto/frameworks/base/tools/aapt2/
DApkInfo.pb.h11007 inline int SupportsInput::inputs_size() const { in inputs_size() function