| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | sink_constant.cc | 55 auto const_op = dyn_cast_or_null<TF::ConstOp>(constant.getDefiningOp()); in runOnFunction() local
56 if (!const_op) return; in runOnFunction()
66 if (constant.use_empty()) const_op.erase(); in runOnFunction()
71 const_op.getOperation()->moveBefore(&body.begin()->front()); in runOnFunction()
74 map_entry.first->getSecond() = const_op.clone(); in runOnFunction()
|
| D | freeze_global_tensors.cc | 111 auto const_op = builder.create<TF::ConstOp>(global_tensor.getLoc(), in runOnOperation() local
115 read_op.getResult().replaceAllUsesWith(const_op.getResult()); in runOnOperation()
|
| D | collection_ops_util.cc | 248 auto const_op = llvm::dyn_cast<TF::ConstOp>(indices_op); in GetFirstIfIndicesAreContiguous() local
249 if (!const_op) return -1; in GetFirstIfIndicesAreContiguous()
252 for (const auto& ind : const_op.value().getValues<APInt>()) { in GetFirstIfIndicesAreContiguous()
|
| D | replicate_to_island.cc | 110 auto const_op = builder.create<TF::ConstOp>( in UpdateRegionReplicateVariantOps() local
114 op->replaceAllUsesWith(const_op); in UpdateRegionReplicateVariantOps()
|
| /external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/ |
| D | legalize_utils.cc | 185 auto const_op = in getTosa1DConstTensorTable() local
187 return const_op.getResult(); in getTosa1DConstTensorTable()
196 auto const_op = in getTosaConstTensorSingleF32() local
198 return const_op.getResult(); in getTosaConstTensorSingleF32()
207 auto const_op = in getTosaConstTensorSingleI32() local
209 return const_op.getResult(); in getTosaConstTensorSingleI32()
359 auto const_op = rewriter.create<T0>(op->getLoc(), const_type, const_attr); in get1DConstTensor() local
360 return const_op.getResult(); in get1DConstTensor()
373 auto const_op = in get1DConstTensor() local
375 return const_op.getResult(); in get1DConstTensor()
[all …]
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | functionalize_cond_test.cc | 124 auto const_op = ops::Const(root.WithOpName("const"), 1); in TEST() local
130 {switch_op.output_false, const_op, const_op}); in TEST()
133 {switch_op.output_true, const_op, const_op}); in TEST()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | prepare_quantize_helper.h | 280 SourceOp op, Operation* const_op, int input_index, in processConstantOp()
284 auto type = const_op->getResult(0).getType().dyn_cast<ShapedType>(); in processConstantOp()
288 if (!matchPattern(const_op->getResult(0), m_Constant(&attr))) { in processConstantOp()
289 const_op->emitError("Not a constant op."); in processConstantOp()
309 llvm::minIntN(10), -llvm::minIntN(10), const_op->getLoc()); in processConstantOp()
319 const_op->emitError("Failed to get quantized type"); in processConstantOp()
324 Type expressed_type = const_op->getResult(0).getType(); in processConstantOp()
326 rewriter.setInsertionPointAfter(const_op); in processConstantOp()
327 auto q = rewriter.create<Q>(const_op->getLoc(), cast_type, in processConstantOp()
328 const_op->getResult(0)); in processConstantOp()
[all …]
|
| D | dilated_conv.h | 125 if (auto const_op = in matchAndRewrite() local
128 (*const_op.value().cast<DenseElementsAttr>().getIntValues().begin()) in matchAndRewrite()
|
| /external/tensorflow/tensorflow/c/eager/parallel_device/ |
| D | parallel_device_lib.h | 240 std::unique_ptr<TFE_Op, decltype(&TFE_DeleteOp)> const_op( in ScalarsFromSequence()
243 TFE_OpSetDevice(const_op.get(), underlying_devices_[device_index].c_str(), in ScalarsFromSequence()
246 TFE_OpSetAttrTensor(const_op.get(), "value", tensor.get(), status); in ScalarsFromSequence()
248 TFE_OpSetAttrType(const_op.get(), "dtype", datatype_enum); in ScalarsFromSequence()
251 TFE_Execute(const_op.get(), &device_handle, &num_outputs, status); in ScalarsFromSequence()
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | export_tensorflow.cc | 162 tensorflow::NodeDef* const_op = tensorflow_graph->add_node(); in ConvertFloatTensorConst() local
163 const_op->set_op("Const"); in ConvertFloatTensorConst()
164 const_op->set_name(name); in ConvertFloatTensorConst()
165 (*const_op->mutable_attr())["dtype"].set_type(DT_FLOAT); in ConvertFloatTensorConst()
166 auto* tensor = (*const_op->mutable_attr())["value"].mutable_tensor(); in ConvertFloatTensorConst()
204 tensorflow::NodeDef* const_op = tensorflow_graph->add_node(); in ConvertFloatTensorConst() local
205 const_op->set_op("Const"); in ConvertFloatTensorConst()
206 const_op->set_name(name); in ConvertFloatTensorConst()
207 (*const_op->mutable_attr())["dtype"].set_type(DT_FLOAT); in ConvertFloatTensorConst()
208 auto* tensor = (*const_op->mutable_attr())["value"].mutable_tensor(); in ConvertFloatTensorConst()
[all …]
|
| /external/tensorflow/tensorflow/compiler/tf2xla/cc/ |
| D | BUILD | 20 "//tensorflow/cc:const_op",
43 "//tensorflow/cc:const_op",
|
| /external/tensorflow/tensorflow/cc/ |
| D | BUILD | 278 name = "const_op",
279 srcs = ["ops/const_op.cc"],
280 hdrs = ["ops/const_op.h"],
298 ":const_op",
581 "ops/const_op.h",
592 ":const_op",
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | remote_execution_test.py | 147 const_op = constant_op.constant(3.0, dtype=dtypes.float32)
150 func=f, inp=[const_op], Tout=dtypes.float32)
|
| /external/angle/third_party/vulkan-deps/spirv-tools/src/source/opt/ |
| D | fold.cpp | 640 const analysis::Constant* const_op = const_mgr->FindDeclaredConstant(id); in FoldInstructionToConstant() local
641 if (!const_op) { in FoldInstructionToConstant()
645 constants.push_back(const_op); in FoldInstructionToConstant()
|
| /external/deqp-deps/SPIRV-Tools/source/opt/ |
| D | fold.cpp | 640 const analysis::Constant* const_op = const_mgr->FindDeclaredConstant(id); in FoldInstructionToConstant() local
641 if (!const_op) { in FoldInstructionToConstant()
645 constants.push_back(const_op); in FoldInstructionToConstant()
|
| /external/swiftshader/third_party/SPIRV-Tools/source/opt/ |
| D | fold.cpp | 640 const analysis::Constant* const_op = const_mgr->FindDeclaredConstant(id); in FoldInstructionToConstant() local
641 if (!const_op) { in FoldInstructionToConstant()
645 constants.push_back(const_op); in FoldInstructionToConstant()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/data/vectorization/ |
| D | BUILD | 18 "//tensorflow/cc:const_op",
|
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | quantize_nodes_test.cc | 160 auto const_op = [&](const string& name, const TensorShape& shape, in TestIgnoreOps() local
171 Output a_op = const_op("a_op", {m, k}, {2}); in TestIgnoreOps()
172 Output b_op = const_op("b_op", {k, n}, {3}); in TestIgnoreOps()
173 Output c_op = const_op("c_op", {m, k}, {1}); in TestIgnoreOps()
174 Output d_op = const_op("d_op", {k, n}, {4}); in TestIgnoreOps()
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/ |
| D | legalize_to_linalg.cc | 892 OpTy const_op, ArrayRef<Value> /*args*/, in matchAndRewrite() argument
894 Location loc = const_op.getLoc(); in matchAndRewrite()
895 auto value_attr = const_op.value().template cast<DenseElementsAttr>(); in matchAndRewrite()
897 ReplaceConstOp(loc, const_op, value_attr, rewriter); in matchAndRewrite()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | BUILD | 28 "const_op.cc",
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_ops_n_z.cc | 254 auto const_op = dyn_cast_or_null<ConstOp>(value.getDefiningOp()); in fold() local
255 if (!const_op) return None; in fold()
257 auto value_attr = const_op.value().dyn_cast<DenseIntElementsAttr>(); in fold()
|
| /external/tensorflow/tensorflow/cc/framework/ |
| D | cc_op_gen.cc | 1069 #include "tensorflow/cc/ops/const_op.h" in StartFiles()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes_test.cc | 172 auto const_op = ops::Const(s.WithOpName(name), t); in MakeConstNodeDef() local
173 return const_op.node()->def(); in MakeConstNodeDef()
|
| /external/tensorflow/tensorflow/ |
| D | tensorflow.bzl | 881 clean_dep("//tensorflow/cc:const_op"),
|