| /external/tensorflow/tensorflow/python/framework/ |
| D | python_api_parameter_converter_test.py | 67 input_specs, argument
72 api_info.InitializeFromParamSpecs(input_specs, attr_specs, param_names,
208 input_specs=dict(x="int32", y="float32"),
216 input_specs=dict(x="T"),
224 input_specs=dict(x="T"),
232 input_specs=dict(x="T", y="T"),
240 input_specs=dict(x="T", y="T"),
250 input_specs=dict(x="T"),
258 input_specs=dict(x="N * T", y="T"),
267 input_specs=dict(x="N * T", y="T"),
[all …]
|
| D | python_api_info_test.py | 53 input_specs, argument
58 api_info.InitializeFromParamSpecs(input_specs, attr_specs, param_names,
247 def testInitializeFromParamSpecs(self, api_name, param_names, input_specs, argument
250 input_specs, attr_specs)
|
| D | python_api_info_wrapper.cc | 39 const std::map<std::string, std::string>& input_specs, in InitializeFromParamSpecs() argument
43 input_specs, attr_specs, param_names, defaults_tuple.ptr()); in InitializeFromParamSpecs()
|
| D | python_api_info.h | 171 const std::map<std::string, std::string>& input_specs,
|
| D | python_api_info.cc | 208 const std::map<std::string, std::string>& input_specs, in InitializeFromParamSpecs() argument
216 for (const auto& input_spec : input_specs) { in InitializeFromParamSpecs()
|
| /external/tensorflow/tensorflow/lite/python/ |
| D | tflite_keras_util.py | 79 …input_specs = model._get_save_spec(dynamic_batch=not keep_original_batch_size) # pylint: disable=…
80 if input_specs is None:
82 input_specs = _enforce_names_consistency(input_specs)
84 if isinstance(input_specs,
85 collections_abc.Sequence) and len(input_specs) == 1:
88 return input_specs
90 return [input_specs]
|
| /external/tensorflow/tensorflow/python/keras/saving/ |
| D | saving_utils.py | 79 …input_specs = model._get_save_spec(dynamic_batch=not keep_original_batch_size) # pylint: disable=…
80 if input_specs is None:
82 input_specs = _enforce_names_consistency(input_specs)
84 if isinstance(input_specs,
85 collections_abc.Sequence) and len(input_specs) == 1:
88 return input_specs
90 return [input_specs]
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/quantization/ |
| D | device_target.cc | 108 llvm::SmallVector<Type, 4> input_specs, out_specs; in DecomposeMultiplyAccumulateScale() local
109 for (auto spec : rop.input_specs()) { in DecomposeMultiplyAccumulateScale()
110 input_specs.push_back(spec.cast<TypeAttr>().getValue()); in DecomposeMultiplyAccumulateScale()
116 auto in_spec = input_specs[0].dyn_cast<quant::UniformQuantizedType>(); in DecomposeMultiplyAccumulateScale()
118 auto w_spec = input_specs[1].dyn_cast<quant::UniformQuantizedType>(); in DecomposeMultiplyAccumulateScale()
119 auto b_spec = input_specs[2].dyn_cast<quant::UniformQuantizedType>(); in DecomposeMultiplyAccumulateScale()
|
| D | quantization_context.cc | 69 signature.reserve(op.input_specs().size() + op.output_specs().size()); in GetSignature()
125 llvm::SmallVector<Attribute, 4> input_specs; in Finalize() local
126 auto original_input_specs = op.input_specs().getValue(); in Finalize()
131 input_specs.push_back(original_input_specs[i]); in Finalize()
133 input_specs.push_back(TypeAttr::get(requantize.params)); in Finalize()
135 input_specs.push_back(TypeAttr::get(state.params)); in Finalize()
138 op->setAttr("input_specs", ArrayAttr::get(context, input_specs)); in Finalize()
286 params_attr = op.input_specs()[index]; in InitializeState()
|
| /external/llvm-project/mlir/test/Dialect/Quant/ |
| D | quant_region.mlir | 10 }) {input_specs = [f32, f32, f32], output_specs = [f32], logical_kernel = "xyz"}
22 }) {input_specs = [!quant.uniform<i8:f32, 1.0>, !quant.uniform<i8:f32, 2.0>, f32],
35 …}) {input_specs = [!quant.uniform<i8:f32, 1.0>, !quant.uniform<i8:f32, 2.0>, !quant.uniform<i32:f3…
50 …}) {input_specs = [!quant.uniform<i8:f32, 1.0>, !quant.uniform<i8:f32, 2.0>, !quant.uniform<i32:f1…
65 }) {input_specs = [!quant.uniform<i8:f32, 1.0>, !quant.uniform<i8:f32, 2.0>, i32],
80 }) {input_specs = [!quant.uniform<i8:f32, 1.0>, !quant.uniform<i8:f32, 2.0>],
96 }) {input_specs = [!quant.uniform<i8:f32, 1.0>, !quant.uniform<i8:f32, 2.0>],
|
| /external/llvm-project/mlir/lib/Dialect/Quant/IR/ |
| D | QuantOps.cpp | 68 if (op.getNumOperands() != op.input_specs().size() || in verifyRegionOp()
74 for (auto input : llvm::zip(op.getOperandTypes(), op.input_specs())) { in verifyRegionOp()
|
| /external/tflite-support/tensorflow_lite_support/cc/task/vision/core/ |
| D | base_vision_task_api.h | 79 ImageTensorSpecs input_specs, in CheckAndSetInputs()
83 if (input_specs.color_space != tflite::ColorSpaceType_RGB) { in CheckAndSetInputs()
89 input_specs_ = absl::make_unique<ImageTensorSpecs>(input_specs); in CheckAndSetInputs()
|
| /external/tensorflow/tensorflow/python/compiler/tensorrt/test/ |
| D | tf_trt_integration_test_base.py | 253 input_specs=[
356 return [spec.name + ":0" for spec in params.input_specs]
404 params.input_specs[i].name: current_input_data[i]
405 for i in range(len(params.input_specs))
434 assert len(params.input_specs) == len(data)
930 for spec in params.input_specs:
961 params.graph_fn, input_signature=params.input_specs)
983 input_specs = self._GetParamsCached().input_specs
985 assert len(input_specs) == len(dim_list)
987 for spec, np_shape in zip(input_specs, dim_list):
|
| D | lru_cache_test.py | 51 input_specs=[
|
| D | dynamic_input_shapes_test.py | 72 input_specs=[
|
| /external/tensorflow/tensorflow/python/saved_model/ |
| D | load_v1_in_v2.py | 141 original_input_names, input_specs = zip(*input_items)
144 input_specs = []
148 for input_spec in input_specs
|
| /external/tensorflow/tensorflow/c/experimental/saved_model/core/ |
| D | saved_model_utils.cc | 159 std::vector<const TensorSpecProto*> input_specs; in ValidateSavedFunctionCompatibleWithFunctionDef() local
160 TF_RETURN_IF_ERROR(FlattenSignature(input_signature, &input_specs)); in ValidateSavedFunctionCompatibleWithFunctionDef()
161 if (input_specs.size() + saved_concrete_function.bound_inputs_size() != in ValidateSavedFunctionCompatibleWithFunctionDef()
166 " inputs, but the SavedConcreteFunction has ", input_specs.size(), in ValidateSavedFunctionCompatibleWithFunctionDef()
|
| /external/tensorflow/tensorflow/python/keras/layers/preprocessing/ |
| D | category_crossing_test.py | 240 input_specs = [
245 output_spec = layer.compute_output_signature(input_specs)
|
| /external/tensorflow/tensorflow/python/keras/saving/saved_model/ |
| D | load.py | 688 input_specs = self._infer_inputs(first_layer)
690 model._set_inputs(input_specs) # pylint: disable=protected-access
691 if not model.built and not isinstance(input_specs, dict):
|
| /external/llvm-project/mlir/include/mlir/Dialect/Quant/ |
| D | QuantOps.td | 100 TypeArrayAttr:$input_specs,
|