Home
last modified time | relevance | path

Searched refs:op_name (Results 1 – 25 of 269) sorted by relevance

1234567891011

/external/tensorflow/tensorflow/core/graph/
Dmkl_graph_util.h132 static inline bool IsMklLayoutDependentOp(const string& op_name, DataType T) { in IsMklLayoutDependentOp() argument
133 string kernel = KernelsRegisteredForOp(op_name); in IsMklLayoutDependentOp()
156 static inline bool IsMklLayoutDependentOp(const string& op_name, in IsMklLayoutDependentOp() argument
158 string kernel = KernelsRegisteredForOp(op_name); in IsMklLayoutDependentOp()
174 static inline bool IsMklNameChangeOp(const string& op_name, DataType T) { in IsMklNameChangeOp() argument
175 string kernel = KernelsRegisteredForOp(op_name); in IsMklNameChangeOp()
210 static inline bool IsMklOp(const string& op_name, DataType T) { in IsMklOp() argument
211 return IsMklLayoutDependentOp(op_name, T) || IsMklNameChangeOp(op_name, T); in IsMklOp()
226 static inline bool IsMklElementWiseOp(const string& op_name, DataType T) { in IsMklElementWiseOp() argument
227 if (!IsMklOp(op_name, T)) { in IsMklElementWiseOp()
[all …]
Dgraph_def_builder.cc101 Node* SourceOp(const string& op_name, const GraphDefBuilder::Options& opts) { in SourceOp() argument
103 NodeBuilder node_builder(opts.GetNameForOp(op_name), op_name, in SourceOp()
108 Node* UnaryOp(const string& op_name, NodeOut input, in UnaryOp() argument
111 NodeBuilder node_builder(opts.GetNameForOp(op_name), op_name, in UnaryOp()
117 Node* BinaryOp(const string& op_name, NodeOut a, NodeOut b, in BinaryOp() argument
120 NodeBuilder node_builder(opts.GetNameForOp(op_name), op_name, in BinaryOp()
/external/tensorflow/tensorflow/compiler/mlir/xla/tests/translate/
Dconditional.hlotxt6 %arg_tuple.7 = (f32[]) parameter(0), metadata={op_name="XLA_Args"}
7 …%get-tuple-element.8 = f32[] get-tuple-element(%arg_tuple.7), index=0, metadata={op_name="XLA_Args…
8 %log.9 = f32[] log(%get-tuple-element.8), metadata={op_type="Log" op_name="cond/Log"}
9 ROOT %tuple.10 = (f32[]) tuple(%log.9), metadata={op_name="XLA_Retvals"}
13 %arg_tuple.12 = (f32[]) parameter(0), metadata={op_name="XLA_Args"}
14 …%get-tuple-element.13 = f32[] get-tuple-element(%arg_tuple.12), index=0, metadata={op_name="XLA_Ar…
15 …%exponential.14 = f32[] exponential(%get-tuple-element.13), metadata={op_type="Exp" op_name="cond/…
16 ROOT %tuple.15 = (f32[]) tuple(%exponential.14), metadata={op_name="XLA_Retvals"}
21 %arg0.1 = f32[] parameter(0), parameter_replication={false}, metadata={op_name="XLA_Args"}
24 %constant.3 = f32[] constant(10), metadata={op_type="Less" op_name="Less"}
[all …]
Dwhile.hlotxt6 %arg_1 = s64[] parameter(0), metadata={op_name="HLO_Args"}
7 …ROOT %compare.2 = pred[] compare(%arg_1, %arg_1), direction=LT, metadata={op_type="Less" op_name="…
11 %arg_1 = s64[] parameter(0), metadata={op_name="HLO_Args"}
12 ROOT %compare.2 = s64[] add(%arg_1, %arg_1), metadata={op_type="Less" op_name="Less"}
17 %arg0.1 = s64[] parameter(0), metadata={op_name="HLO_Args"}
/external/tensorflow/tensorflow/core/ops/
Dimage_ops_test.cc31 for (const char* op_name : {"ResizeArea", "ResizeBicubic", "ResizeBilinear", in TEST()
33 ShapeInferenceTestOp op(op_name); in TEST()
65 for (const char* op_name : {"DecodeJpeg", "DecodePng"}) { in TEST()
66 ShapeInferenceTestOp op(op_name); in TEST()
72 TF_ASSERT_OK(NodeDefBuilder("test", op_name) in TEST()
78 TF_ASSERT_OK(NodeDefBuilder("test", op_name) in TEST()
85 TF_ASSERT_OK(NodeDefBuilder("test", op_name) in TEST()
94 const char* op_name = "DecodeAndCropJpeg"; in TEST() local
95 ShapeInferenceTestOp op(op_name); in TEST()
104 TF_ASSERT_OK(NodeDefBuilder("test", op_name) in TEST()
[all …]
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/
Dderived_attr_populator_gen.cc49 auto op_name = op.getCppClassName(); in EmitOpAttrPopulators() local
50 OUT(0) << "static Status Populate" << op_name in EmitOpAttrPopulators()
51 << "DerivedAttrs(mlir::TF::" << op_name in EmitOpAttrPopulators()
94 auto op_name = op.getCppClassName(); in EmitInstAttrPopulator() local
97 OUT(2) << "if (auto tfOp = llvm::dyn_cast<mlir::TF::" << op_name in EmitInstAttrPopulator()
99 OUT(4) << "TF_RETURN_IF_ERROR(Populate" << op_name in EmitInstAttrPopulator()
115 auto op_name = op.getCppClassName(); in EmitOpAttrNameCollector() local
116 OUT(0) << "static void Collect" << op_name in EmitOpAttrNameCollector()
117 << "DerivedAttrsName(mlir::TF::" << op_name in EmitOpAttrNameCollector()
144 auto op_name = op.getCppClassName(); in EmitInstAttrNameCollector() local
[all …]
/external/tensorflow/tensorflow/c/
Dkernels_test.cc94 const char* op_name, in GetFakeKernel() argument
97 def.set_op(op_name); in GetFakeKernel()
114 const char* op_name = "FooOp"; in TEST() local
117 REGISTER_OP(op_name) in TEST()
124 op_name, device_name, &MyCreateFunc, &MyComputeFunc, &MyDeleteFunc); in TEST()
130 TF_Buffer* buf = TF_GetRegisteredKernelsForOp(op_name, status); in TEST()
143 GetFakeKernel(device_name, op_name, &status); in TEST()
166 const char* op_name = "BarOp"; in TEST() local
169 REGISTER_OP(op_name) in TEST()
210 TF_KernelBuilder* builder = TF_NewKernelBuilder(op_name, device_name, nullptr, in TEST()
[all …]
/external/tensorflow/tensorflow/lite/experimental/writer/
Doption_writer_generator.cc199 for (auto op_name : ops_) { in BuildOpToOptionMap() local
203 ToCollapsed(op_name) + "options"; in BuildOpToOptionMap()
209 op_to_option_.insert(std::make_pair(op_name, option_name)); in BuildOpToOptionMap()
214 auto it = op_to_option_.find(op_name); in BuildOpToOptionMap()
216 std::cerr << "Didn't find option for " << op_name << std::endl; in BuildOpToOptionMap()
267 void GenerateImportForOp(FILE* fp, const std::string& op_name, in GenerateImportForOp() argument
278 fprintf(fp, " case BuiltinOperator_%s: {\n", op_name.c_str()); in GenerateImportForOp()
368 for (const auto& op_name : option->ops()) { in GenerateImport() local
369 auto option_it = option->op_to_option().find(op_name); in GenerateImport()
370 if (!option_it->second.empty() && ignores.find(op_name) == ignores.end()) in GenerateImport()
[all …]
/external/tensorflow/tensorflow/lite/toco/logging/
Dconversion_log_util.cc38 string op_name; in TryGetOperatorName() local
45 op_name = node_def.op(); in TryGetOperatorName()
46 if (!op_name.empty()) return op_name; in TryGetOperatorName()
57 op_name = unsupported_op.tensorflow_op; in TryGetOperatorName()
58 return op_name; in TryGetOperatorName()
62 op_name = OperatorTypeName(op.type); in TryGetOperatorName()
63 return op_name; in TryGetOperatorName()
159 const string op_name = TryGetOperatorName(*op); in CountOperatorsByType() local
161 (*custom_ops)[op_name]++; in CountOperatorsByType()
163 (*select_ops)[op_name]++; in CountOperatorsByType()
[all …]
/external/ltp/testcases/kernel/syscalls/statx/
Dstatx06.c103 char *op_name; member
106 .op_name = "Birth time"},
108 .op_name = "Modified time"},
110 .op_name = "Access time"},
112 .op_name = "Change time"}
152 tst_res(TFAIL, "%s < before time", tc->op_name); in test_statx()
154 tst_res(TFAIL, "%s > after_time", tc->op_name); in test_statx()
156 tst_res(TPASS, "%s Passed", tc->op_name); in test_statx()
/external/tensorflow/tensorflow/compiler/jit/
Dcompilability_check_util.h234 bool IsAssertOrCheckNumerics(absl::string_view op_name) const { in IsAssertOrCheckNumerics() argument
235 return op_name == "Assert" || op_name == "CheckNumerics"; in IsAssertOrCheckNumerics()
238 bool IsStatefulRandomOp(absl::string_view op_name) const { in IsStatefulRandomOp() argument
239 return op_name == "RandomUniform" || op_name == "RandomShuffle" || in IsStatefulRandomOp()
240 op_name == "RandomUniformInt" || op_name == "RandomStandardNormal" || in IsStatefulRandomOp()
241 op_name == "TruncatedNormal" || op_name == "Multinomial"; in IsStatefulRandomOp()
/external/tensorflow/tensorflow/core/framework/
Ddataset_stateful_op_whitelist.h28 Status Add(string op_name) { in Add() argument
29 op_names_.insert(std::move(op_name)); in Add()
33 Status Remove(string op_name) { in Remove() argument
34 op_names_.erase(op_name); in Remove()
38 bool Contains(const string& op_name) { return op_names_.count(op_name); } in Contains() argument
/external/tensorflow/tensorflow/compiler/mlir/lite/
Doperator_converter_gen.cc59 static inline std::string GetOperatorBuilderName(StringRef op_name) { in GetOperatorBuilderName() argument
60 assert(op_name.startswith("TFL_") && "unexpected op prefix"); in GetOperatorBuilderName()
61 assert(op_name.endswith("Op") && "unexpected op suffix"); in GetOperatorBuilderName()
65 oss << "Create" << op_name.drop_front(4).str() << "erator"; in GetOperatorBuilderName()
79 StringRef op_name = def->getName().drop_front(4); // Strip 'TFL_' prefix in EmitOptionBuilders() local
85 << option_name << "(mlir::TFL::" << op_name in EmitOptionBuilders()
149 StringRef op_name = def->getName().drop_front(4); in EmitOperatorBuilders() local
153 << GetOperatorBuilderName(def->getName()) << "(mlir::TFL::" << op_name in EmitOperatorBuilders()
210 StringRef op_name = def->getName().drop_front(4); in EmitGetBuiltinOpCode() local
211 os << " if (isa<mlir::TFL::" << op_name << ">(op))\n" in EmitGetBuiltinOpCode()
[all …]
/external/v8/src/codegen/s390/
Dassembler-s390.h336 #define DECLARE_S390_RIL_AB_INSTRUCTIONS(name, op_name, op_value) \ argument
339 ril_format(op_name, r1.code(), i2.immediate()); \
341 #define DECLARE_S390_RIL_C_INSTRUCTIONS(name, op_name, op_value) \ argument
343 ril_format(op_name, m1, i2.immediate()); \
359 #define DECLARE_S390_RR_INSTRUCTIONS(name, op_name, op_value) \ in S390_RIL_B_OPCODE_LIST() argument
361 rr_format(op_name, r1.code(), r2.code()); \ in S390_RIL_B_OPCODE_LIST()
364 rr_format(op_name, r1.code(), r2.code()); \
367 rr_format(op_name, m1, r2.code()); \
378 #define DECLARE_S390_RRD_INSTRUCTIONS(name, op_name, op_value) \ in S390_RR_OPCODE_LIST() argument
381 rrd_format(op_name, r1.code(), r3.code(), r2.code()); \ in S390_RR_OPCODE_LIST()
[all …]
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dmkl_eager_op_rewrite.cc31 string op_name; member
73 bool FastCheckIfKernelRegistered(std::string op_name, DataType dt);
77 bool SlowCheckIfKernelRegistered(std::string op_name, DataType dt);
100 mkl_eager_ops_.insert(std::make_pair(op.op_name, op)); in InsertMKLEagerOps()
189 bool MklEagerOpRewrite::FastCheckIfKernelRegistered(std::string op_name, in FastCheckIfKernelRegistered() argument
193 string registered_kernels_key = op_name + std::to_string(dt); in FastCheckIfKernelRegistered()
199 kernel_registered = SlowCheckIfKernelRegistered(op_name, dt); in FastCheckIfKernelRegistered()
210 bool MklEagerOpRewrite::SlowCheckIfKernelRegistered(string op_name, in SlowCheckIfKernelRegistered() argument
213 auto element = mkl_eager_ops_.find(op_name); in SlowCheckIfKernelRegistered()
217 mkl_op_registry::GetMklEagerOpName(op_name), dt) || in SlowCheckIfKernelRegistered()
[all …]
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
Dexport_utils.cc230 auto op_name = op_name_or_status.ConsumeValueOrDie(); in IsRefTypeControlOp() local
231 if (op_name.equals("NextIteration")) in IsRefTypeControlOp()
235 if (op_name.equals("Enter") || op_name.equals("Exit") || in IsRefTypeControlOp()
236 op_name.equals("Switch") || op_name.equals("Merge")) { in IsRefTypeControlOp()
245 StatusOr<llvm::StringRef> GetTensorFlowOpName(llvm::StringRef op_name) { in GetTensorFlowOpName() argument
253 if (!op_name.consume_front("_tf.") && !op_name.consume_front("tf.") && in GetTensorFlowOpName()
254 !op_name.consume_front("tf_executor.")) { in GetTensorFlowOpName()
255 return errors::FailedPrecondition("op node '", op_name.str(), in GetTensorFlowOpName()
260 if (!op_name.consume_back(".sink")) op_name.consume_back(".Sink"); in GetTensorFlowOpName()
261 return op_name; in GetTensorFlowOpName()
[all …]
/external/tensorflow/tensorflow/tools/graph_transforms/
Dremove_attribute.cc40 string op_name; in RemoveAttribute() local
47 op_name = context.params.at("op_name")[0]; in RemoveAttribute()
49 op_name = "*"; in RemoveAttribute()
57 if (((op_name == "*") || (op_name == node.op())) && in RemoveAttribute()
Drename_attribute.cc42 string op_name; in RenameAttribute() local
44 op_name = context.params.at("op_name")[0]; in RenameAttribute()
46 op_name = "*"; in RenameAttribute()
55 if (((op_name == "*") || (op_name == node.op())) && in RenameAttribute()
/external/tensorflow/tensorflow/python/debug/lib/
Ddebug_events_reader.py402 if graph_op_creation_digest.op_name in self._op_by_name:
405 (graph_op_creation_digest.op_name, graph_op_creation_digest.op_type))
407 graph_op_creation_digest.op_name] = graph_op_creation_digest
425 def get_op_type(self, op_name): argument
426 return self._op_by_name[op_name].op_type
428 def get_tensor_id(self, op_name, output_slot): argument
430 return self._op_by_name[op_name].output_tensor_ids[output_slot]
482 op_name, argument
489 self._op_name = op_name
503 def op_name(self): member in GraphOpCreationDigest
[all …]
/external/tensorflow/tensorflow/python/eager/
Dforwardprop.py71 def _jvp_helper(op_name, attr_tuple, inputs, outputs, tangents): argument
92 _TRACE_COUNT[op_name] = _TRACE_COUNT.get(op_name, 0) + 1
94 special_case = _SPECIAL_CASES.get(op_name, None)
116 execute.record_gradient(op_name, inputs, attr_tuple, outputs)
158 def _jvp_dispatch(op_name, attr_tuple, inputs, outputs, tangents): argument
162 if _TRACE_COUNT.get(op_name, 0) < _TRACE_COUNT_LIMIT:
164 op_name, attr_tuple, inputs, outputs, tangents)
167 op_name, attr_tuple, inputs, outputs, tangents)
/external/tensorflow/tensorflow/core/grappler/
Dop_types.cc646 const string& op_name = node.op(); in IsStateful() local
647 Status status = op_registry->LookUpOpDef(op_name, &op_def); in IsStateful()
649 LOG(WARNING) << "Failed to lookup OpDef for " << op_name in IsStateful()
667 const string& op_name = node.op(); in IsFreeOfSideEffect() local
668 Status status = op_registry->LookUpOpDef(op_name, &op_def); in IsFreeOfSideEffect()
698 string op_name = node.op(); in ModifiesInputsInPlace() local
701 if (op_name == "AssignVariableOp" || op_name == "AssignAddVariableOp" || in ModifiesInputsInPlace()
702 op_name == "AssignSubVariableOp" || op_name == "ResourceScatterUpdate" || in ModifiesInputsInPlace()
703 op_name == "ResourceScatterAdd" || op_name == "ResourceScatterSub" || in ModifiesInputsInPlace()
704 op_name == "ResourceScatterMul" || op_name == "ResourceScatterDiv" || in ModifiesInputsInPlace()
[all …]
/external/tensorflow/tensorflow/lite/experimental/ruy/
Dcheck_macros_test.cc29 #define TEST_COMPARISON_FOR_FAMILY(family, vacuously_succeeds, op_name, x, op, \ argument
33 RUY_##family##_##op_name(x, y); \
42 #define TEST_COMPARISON(op_name, x, op, y) \ argument
44 TEST_COMPARISON_FOR_FAMILY(CHECK, false, op_name, x, op, y); \
52 #define TEST_COMPARISON(op_name, x, op, y) \
54 TEST_COMPARISON_FOR_FAMILY(CHECK, false, op_name, x, op, y); \
55 TEST_COMPARISON_FOR_FAMILY(DCHECK, false, op_name, x, op, y); \
/external/tensorflow/tensorflow/python/framework/
Dc_api_util.py129 def get_api_def(self, op_name): argument
131 buf = c_api.TF_ApiDefMapGet(self._api_def_map, op_name, len(op_name))
138 def get_op_def(self, op_name): argument
139 if op_name in self._op_per_name:
140 return self._op_per_name[op_name]
141 raise ValueError("No entry found for " + op_name + ".")
/external/tensorflow/tensorflow/core/grappler/optimizers/data/
Dmake_sloppy.cc48 for (const auto& op_name : kSloppyAttrOps) { in OptimizeAndCollectStats() local
49 if (node.op() == op_name) { in OptimizeAndCollectStats()
55 for (const auto& op_name : kDeterministicAttrOps) { in OptimizeAndCollectStats() local
56 if (node.op() == op_name && in OptimizeAndCollectStats()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dscoped_allocator_optimizer.cc50 bool HasOpName(const string& node_name, const string& op_name) { in HasOpName() argument
70 return node_name.substr(begin, end - begin) == op_name; in HasOpName()
636 const string& op_name, const string& sac_name, in BuildReplacementOp() argument
639 NodeDefBuilder op_builder(sa_op_name, op_name); in BuildReplacementOp()
665 const string& op_name, int sa_id, in BuildSplitNode() argument
699 const string& op_name, const string& sas_name) { in RewireSubgraph() argument
757 if (!HasOpName(n->name(), op_name)) { in RewireSubgraph()
761 << " " << op_name << " not in " << n->name(); in RewireSubgraph()
795 GraphDef* graph, const string& op_name, in Rewrite() argument
803 VLOG(1) << "UnaryElementwiseRewriter::Rewrite " << op_name in Rewrite()
[all …]

1234567891011