Searched refs:output_specs (Results 1 – 9 of 9) sorted by relevance
| /external/llvm-project/mlir/test/Dialect/Quant/ |
| D | quant_region.mlir | 10 }) {input_specs = [f32, f32, f32], output_specs = [f32], logical_kernel = "xyz"}
23 output_specs = [!quant.uniform<i8:f32, 4.0>], logical_kernel = "xyz"}
36 output_specs = [!quant.uniform<i8:f32, 4.0>], logical_kernel = "xyz"}
51 output_specs = [!quant.uniform<i8:f32, 4.0>], logical_kernel = "xyz"}
66 output_specs = [!quant.uniform<i8:f32, 4.0>], logical_kernel = "xyz"}
81 output_specs = [!quant.uniform<i8:f32, 4.0>], logical_kernel = "xyz"}
97 output_specs = [!quant.uniform<i8:f32, 4.0>], logical_kernel = "xyz"}
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/quantization/ |
| D | quantization_context.cc | 69 signature.reserve(op.input_specs().size() + op.output_specs().size()); in GetSignature()
140 llvm::SmallVector<Attribute, 4> output_specs; in Finalize() local
141 auto original_output_specs = op.output_specs().getValue(); in Finalize()
146 output_specs.push_back(original_output_specs[res]); in Finalize()
148 output_specs.push_back(TypeAttr::get(requantize.params)); in Finalize()
150 output_specs.push_back(TypeAttr::get(state.params)); in Finalize()
153 op->setAttr("output_specs", ArrayAttr::get(context, output_specs)); in Finalize()
284 params_attr = op.output_specs()[index]; in InitializeState()
|
| D | device_target.cc | 112 for (auto spec : rop.output_specs()) { in DecomposeMultiplyAccumulateScale()
162 auto o_spec = rop.output_specs()[0] in DecomposeSameScale()
|
| /external/llvm-project/mlir/lib/Dialect/Quant/IR/ |
| D | QuantOps.cpp | 69 op.getNumResults() != op.output_specs().size()) in verifyRegionOp()
83 for (auto result : llvm::zip(op.getResultTypes(), op.output_specs())) { in verifyRegionOp()
|
| /external/tensorflow/tensorflow/c/experimental/saved_model/core/ |
| D | saved_model_utils.cc | 173 std::vector<const TensorSpecProto*> output_specs; in ValidateSavedFunctionCompatibleWithFunctionDef() local
174 TF_RETURN_IF_ERROR(FlattenSignature(output_signature, &output_specs)); in ValidateSavedFunctionCompatibleWithFunctionDef()
175 if (output_specs.size() != function_def->signature().output_arg_size()) { in ValidateSavedFunctionCompatibleWithFunctionDef()
179 " outputs, but the SavedConcreteFunction has ", output_specs.size(), in ValidateSavedFunctionCompatibleWithFunctionDef()
|
| /external/tensorflow/tensorflow/python/compiler/tensorrt/test/ |
| D | lru_cache_test.py | 54 output_specs=[
|
| D | dynamic_input_shapes_test.py | 76 output_specs=[
|
| D | tf_trt_integration_test_base.py | 257 output_specs=[
361 return [spec.name + ":0" for spec in params.output_specs]
|
| /external/llvm-project/mlir/include/mlir/Dialect/Quant/ |
| D | QuantOps.td | 101 TypeArrayAttr:$output_specs,
|