Home
last modified time | relevance | path

Searched defs:input_spec (Results 1 – 15 of 15) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/python/keras/engine/
H A Dinput_spec.py155 def assert_input_compatibility(input_spec, inputs, layer_name): argument
277 def to_tensor_spec(input_spec, default_dtype=None): argument
H A Dsequential.py505 def input_spec(self): member in Sequential
513 def input_spec(self, value): member in Sequential
H A Dfunctional.py263 def input_spec(self): member in Functional
285 def input_spec(self, value): member in Functional
H A Dbase_layer_v1.py896 def input_spec(self): member in Layer
903 def input_spec(self, value): member in Layer
H A Dbase_layer.py1252 def input_spec(self): member in Layer
1290 def input_spec(self, value): member in Layer
/aosp_15_r20/external/pytorch/test/mobile/nnc/
H A Dtest_context.cpp39 InputSpec input_spec; in create_test_input_spec() local
/aosp_15_r20/external/pytorch/torch/distributed/tensor/
H A D_op_schema.py104 def input_spec(self, index: int = 0) -> DTensorSpec: member in PlacementStrategy
/aosp_15_r20/external/pytorch/torch/csrc/jit/mobile/nnc/
H A Dcontext.cpp173 for (const auto& input_spec : input_specs_) { in serialize() local
/aosp_15_r20/external/tensorflow/tensorflow/python/framework/
H A Dpython_api_info.cc216 for (const auto& input_spec : input_specs) { in InitializeFromParamSpecs() local
/aosp_15_r20/external/tensorflow/tensorflow/c/
H A Dops.cc45 const char* input_spec) { in TF_OpDefinitionBuilderAddInput()
/aosp_15_r20/external/cronet/url/
H A Dgurl.cc74 void GURL::InitCanonical(T input_spec, bool trim_path_end) { in InitCanonical()
/aosp_15_r20/external/tensorflow/tensorflow/core/runtime_fallback/kernel/
H A Dtfrt_op_kernel.cc225 TFRTOpMetaBuilder& TFRTOpMetaBuilder::Input(StringPiece input_spec) { in Input()
/aosp_15_r20/external/tensorflow/tensorflow/lite/experimental/acceleration/configuration/
H A Dflatbuffer_to_proto_test.cc367 EdgeTpuDeviceSpecT* input_spec = input_settings->edgetpu_device_spec.get(); in TEST_F() local
/aosp_15_r20/external/mesa3d/src/amd/compiler/tests/
H A Dhelpers.cpp92 setup_cs(const char* input_spec, enum amd_gfx_level gfx_level, enum radeon_family family, in setup_cs()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/python/
H A Dpmap_lib.cc111 const InputSpec& input_spec, py::handle py_devices, in ShardArg()