Searched refs:div_op (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/lite/delegates/hexagon/builders/ |
| D | rsqrt_builder.cc | 136 auto* div_op = graph_builder_->AddNode(GetTFLiteNodeID()); in PopulateSubGraph() local
137 div_op->SetOpType(OP_QuantizedDiv_8); in PopulateSubGraph()
138 div_op->AddInput(numerator_out_); in PopulateSubGraph()
139 div_op->AddInput(reshape_out); in PopulateSubGraph()
140 div_op->AddInput(numerator_min_); in PopulateSubGraph()
141 div_op->AddInput(numerator_max_); in PopulateSubGraph()
142 div_op->AddInput(sqrt_output_min); in PopulateSubGraph()
143 div_op->AddInput(sqrt_output_max); in PopulateSubGraph()
144 div_op->AddInput(TensorID(output_min_const->GetID(), 0)); in PopulateSubGraph()
145 div_op->AddInput(TensorID(output_max_const->GetID(), 0)); in PopulateSubGraph()
[all …]
|
| /external/tensorflow/tensorflow/c/experimental/ops/ |
| D | math_ops.cc | 133 AbstractOperationPtr div_op(ctx->CreateOperation()); in Div() local
134 TF_RETURN_IF_ERROR(div_op->Reset("Div", /*raw_device_name=*/nullptr)); in Div()
135 TF_RETURN_IF_ERROR(MaybeSetOpName(div_op.get(), name)); in Div()
136 TF_RETURN_IF_ERROR(div_op->AddInput(inputs[0])); // x in Div()
137 TF_RETURN_IF_ERROR(div_op->AddInput(inputs[1])); // y in Div()
140 TF_RETURN_IF_ERROR(div_op->Execute(outputs, &num_retvals)); // z = x / y in Div()
147 AbstractOperationPtr div_op(ctx->CreateOperation()); in DivNoNan() local
148 TF_RETURN_IF_ERROR(div_op->Reset("DivNoNan", /*raw_device_name=*/nullptr)); in DivNoNan()
149 TF_RETURN_IF_ERROR(MaybeSetOpName(div_op.get(), name)); in DivNoNan()
150 TF_RETURN_IF_ERROR(div_op->AddInput(inputs[0])); // x in DivNoNan()
[all …]
|
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/tests/ |
| D | identify_l2_normalization_test.cc | 64 auto div_op = new DivOperator; in RunIdentifyL2Normalization() local
65 div_op->inputs = {"input0", sqrt_op->outputs[0]}; in RunIdentifyL2Normalization()
66 div_op->outputs = {"Divoutput"}; in RunIdentifyL2Normalization()
69 model.operators.push_back(std::unique_ptr<Operator>(div_op)); in RunIdentifyL2Normalization()
|
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | unpartition_embedding_lookup.cc | 166 Operator* div_op = GetOpWithOutput(*model, data_partition_op->inputs[0]); in Run() local
168 CHECK(div_op && div_op->type == OperatorType::kFloorDiv) in Run()
227 LogName(*div_op), LogName(*mod_op), LogName(*stitch_op), in Run()
236 DeleteOpAndArrays(model, div_op); in Run()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | legalize_hlo.cc | 772 mhlo::DivOp div_op, ArrayRef<Value> args, in matchAndRewrite() argument
775 dyn_cast_or_null<mhlo::ReduceWindowOp>(div_op.lhs().getDefiningOp()); in matchAndRewrite()
790 if (div_op.getType() != rw.getType()) return failure(); in matchAndRewrite()
839 if (matchPattern(div_op.rhs(), m_Constant(&divisor))) { in matchAndRewrite()
855 div_op, rw.operand(), in matchAndRewrite()
861 dyn_cast_or_null<mhlo::ReduceWindowOp>(div_op.rhs().getDefiningOp()); in matchAndRewrite()
886 div_op, rw.operand(), in matchAndRewrite()
909 div_op, rw.operand(), in matchAndRewrite()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | reduction_gpu_kernels.cu.h | 1182 DividesBy<T> div_op(static_cast<T>(divisor));
1183 TransformOutputIterator<T, T, DividesBy<T>> itr((T*)out.data(), div_op);
1217 DividesBy<float, Eigen::half> div_op(divisor);
1226 outputIterType itr((Eigen::half*)out.data(), div_op);
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | export_tensorflow.cc | 719 tensorflow::NodeDef* div_op = tensorflow_graph->add_node(); in ConvertDivOperator() local
720 div_op->set_op("Div"); in ConvertDivOperator()
721 div_op->set_name(src_op.outputs[0]); in ConvertDivOperator()
723 *div_op->add_input() = src_op.inputs[0]; in ConvertDivOperator()
724 *div_op->add_input() = src_op.inputs[1]; in ConvertDivOperator()
725 (*div_op->mutable_attr())["T"].set_type( in ConvertDivOperator()
|