| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ops/ |
| 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
|
| D | binary_ops.cc | 144 static constexpr std::array<InputArgSpec, 2> InputSpec() { in InputSpec() function in tensorflow::tensorrt::convert::__anon4744500a0111::ConvertBinaryImpl
168 static constexpr std::array<InputArgSpec, 2> InputSpec() { in InputSpec() function in tensorflow::tensorrt::convert::__anon4744500a0111::ConvertBinary
201 static constexpr std::array<InputArgSpec, 2> InputSpec() { in InputSpec() function in tensorflow::tensorrt::convert::__anon4744500a0111::ConvertBooleanBinary
|
| 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
|
| D | log_softmax.cc | 35 static constexpr std::array<InputArgSpec, 1> InputSpec() { in InputSpec() function in tensorflow::tensorrt::convert::ConvertLogSoftmax
|
| 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
|
| 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
|
| D | like_ops.cc | 38 static constexpr std::array<InputArgSpec, 1> InputSpec() { in InputSpec() function in tensorflow::tensorrt::convert::ConvertLikeOps
|
| D | data_format_vec_permute.cc | 48 static constexpr std::array<InputArgSpec, 1> InputSpec() { in InputSpec() function in tensorflow::tensorrt::convert::ConvertDataFormatVecPermute
|
| D | tile.cc | 35 static constexpr std::array<InputArgSpec, 2> InputSpec() { in InputSpec() function in tensorflow::tensorrt::convert::ConvertTile
|
| D | einsum.cc | 674 static constexpr std::array<InputArgSpec, 2> InputSpec() { in InputSpec() function in tensorflow::tensorrt::convert::__anon861127650111::ConvertEinsum
|
| /external/tensorflow/tensorflow/python/layers/ |
| D | base.py | 18 InputSpec = base.InputSpec variable
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | op_converter_test.cc | 47 static constexpr std::array<InputArgSpec, 2> InputSpec() { in InputSpec() function in tensorflow::tensorrt::convert::ExampleOpConverter
|
| /external/tensorflow/tensorflow/python/keras/engine/ |
| D | input_spec.py | 30 class InputSpec(object): class
|
| D | base_layer_v1.py | 2408 InputSpec = input_spec.InputSpec # pylint:disable=invalid-name variable
|
| D | base_layer.py | 3270 InputSpec = input_spec.InputSpec # pylint:disable=invalid-name variable
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | pmap_lib.cc | 57 struct InputSpec { struct
58 InputSpec(ShardingSpec sharding_spec, py::object indices) in InputSpec() function
60 ShardingSpec sharding_spec;
61 py::object indices;
|
| /external/swiftshader/third_party/llvm-16.0/llvm/lib/Analysis/ |
| D | TFLiteUtils.cpp | 152 auto &InputSpec = InputSpecs[I]; in TFModelEvaluatorImpl() local
|
| /external/tensorflow/tensorflow/python/keras/legacy_tf_layers/ |
| D | base.py | 38 InputSpec = base_layer.InputSpec # pylint: disable=invalid-name variable
|