| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ops/ |
| H A D | unary_ops.cc | 126 static constexpr std::array<InputArgSpec, 1> InputSpec() { in InputSpec() function in tensorflow::tensorrt::convert::ConvertUnaryImpl
146 static constexpr std::array<InputArgSpec, 1> InputSpec() { in InputSpec() function in tensorflow::tensorrt::convert::ConvertUnary
166 static constexpr std::array<InputArgSpec, 1> InputSpec() { in InputSpec() function in tensorflow::tensorrt::convert::ConvertBooleanUnary
193 static constexpr std::array<InputArgSpec, 1> InputSpec() { in InputSpec() function in tensorflow::tensorrt::convert::ConvertActivation
|
| H A D | quantization_ops.cc | 117 static constexpr std::array<InputArgSpec, 3> InputSpec() { in InputSpec() function
232 static constexpr std::array<InputArgSpec, 4> InputSpec() { in InputSpec() function
261 static constexpr std::array<InputArgSpec, 3> InputSpec() { in InputSpec() function
287 static constexpr std::array<InputArgSpec, 1> InputSpec() { in InputSpec() function
367 static constexpr auto InputSpec() { return QDQOpSpec<TFOpType>::InputSpec(); } in InputSpec() function in tensorflow::tensorrt::convert::ConvertQDQ
|
| H A D | binary_ops.cc | 144 static constexpr std::array<InputArgSpec, 2> InputSpec() { in InputSpec() function in tensorflow::tensorrt::convert::__anon4e53edec0111::ConvertBinaryImpl
168 static constexpr std::array<InputArgSpec, 2> InputSpec() { in InputSpec() function in tensorflow::tensorrt::convert::__anon4e53edec0111::ConvertBinary
201 static constexpr std::array<InputArgSpec, 2> InputSpec() { in InputSpec() function in tensorflow::tensorrt::convert::__anon4e53edec0111::ConvertBooleanBinary
|
| H A D | fill_ops.cc | 53 static constexpr std::array<InputArgSpec, 2> InputSpec() { in InputSpec() function in tensorflow::tensorrt::convert::ConvertFill
120 static constexpr std::array<InputArgSpec, 3> InputSpec() { in InputSpec() function in tensorflow::tensorrt::convert::ConvertRange
|
| H A D | variable_ops.cc | 151 static constexpr std::array<InputArgSpec, 0> InputSpec() { return {}; } in InputSpec() function in tensorflow::tensorrt::convert::ConvertVariableV2
253 static constexpr std::array<InputArgSpec, 1> InputSpec() { in InputSpec() function in tensorflow::tensorrt::convert::ConvertReadVariableOp
|
| H A D | log_softmax.cc | 35 static constexpr std::array<InputArgSpec, 1> InputSpec() { in InputSpec() function in tensorflow::tensorrt::convert::ConvertLogSoftmax
|
| H A D | like_ops.cc | 38 static constexpr std::array<InputArgSpec, 1> InputSpec() { in InputSpec() function in tensorflow::tensorrt::convert::ConvertLikeOps
|
| H A D | data_format_vec_permute.cc | 48 static constexpr std::array<InputArgSpec, 1> InputSpec() { in InputSpec() function in tensorflow::tensorrt::convert::ConvertDataFormatVecPermute
|
| H A D | tile.cc | 35 static constexpr std::array<InputArgSpec, 2> InputSpec() { in InputSpec() function in tensorflow::tensorrt::convert::ConvertTile
|
| H A D | einsum.cc | 674 static constexpr std::array<InputArgSpec, 2> InputSpec() { in InputSpec() function in tensorflow::tensorrt::convert::__anonc98924470111::ConvertEinsum
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/python/ |
| H A D | pmap_lib.cc | 57 struct InputSpec { struct
58 InputSpec(ShardingSpec sharding_spec, py::object indices) in InputSpec() argument
60 ShardingSpec sharding_spec;
61 py::object indices;
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/mobile/nnc/ |
| H A D | context.h | 18 struct TORCH_API InputSpec { struct
30 std::vector<int64_t> sizes_;
|
| H A D | context.cpp | 29 InputSpec::InputSpec(const c10::IValue& value) { in InputSpec() function in torch::jit::mobile::nnc::InputSpec
|
| /aosp_15_r20/external/pigweed/pw_sensor/py/pw_sensor/ |
| H A D | constants_generator.py | 652 class InputSpec: class
670 spec: InputSpec,
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/layers/ |
| H A D | base.py | 18 InputSpec = base.InputSpec variable
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| H A D | op_converter_test.cc | 47 static constexpr std::array<InputArgSpec, 2> InputSpec() { in InputSpec() function in tensorflow::tensorrt::convert::ExampleOpConverter
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/keras/engine/ |
| H A D | input_spec.py | 30 class InputSpec(object): class
|
| /aosp_15_r20/external/swiftshader/third_party/llvm-16.0/llvm/lib/Analysis/ |
| H A D | TFLiteUtils.cpp | 152 auto &InputSpec = InputSpecs[I]; in TFModelEvaluatorImpl() local
|
| /aosp_15_r20/external/executorch/exir/ |
| H A D | lowered_backend_module.py | 847 input_specs_to_delete: Dict[str, InputSpec], argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/keras/legacy_tf_layers/ |
| H A D | base.py | 38 InputSpec = base_layer.InputSpec # pylint: disable=invalid-name variable
|
| /aosp_15_r20/external/pytorch/torch/_export/serde/ |
| H A D | schema.py | 265 class InputSpec(_Union): class
|
| H A D | serialize.py | 945 def serialize_input_spec(self, spec: ep.InputSpec) -> InputSpec:
1738 def deserialize_input_spec(self, i: InputSpec) -> ep.InputSpec:
|
| /aosp_15_r20/external/executorch/exir/serde/ |
| H A D | schema.py | 272 class InputSpec(_Union): class
|
| H A D | export_serialize.py | 915 def serialize_input_spec(self, spec: ep.InputSpec) -> InputSpec:
1715 def deserialize_input_spec(self, i: InputSpec) -> ep.InputSpec:
|
| /aosp_15_r20/external/pytorch/torch/export/ |
| H A D | graph_signature.py | 74 class InputSpec: class
|