Home
last modified time | relevance | path

Searched refs:TryGetNodeAttr (Results 1 – 25 of 29) sorted by relevance

12

/external/tensorflow/tensorflow/core/framework/
Dnode_def_util.h250 bool TryGetNodeAttr(const AttrSlice& attrs, StringPiece attr_name,
257 bool TryGetNodeAttr(const AttrSlice& attrs, StringPiece attr_name,
269 bool TryGetNodeAttr(const AttrSlice& attrs, StringPiece attr_name,
271 bool TryGetNodeAttr(const AttrSlice& attrs, StringPiece attr_name,
273 bool TryGetNodeAttr(const AttrSlice& attrs, StringPiece attr_name,
275 bool TryGetNodeAttr(const AttrSlice& attrs, StringPiece attr_name,
277 bool TryGetNodeAttr(const AttrSlice& attrs, StringPiece attr_name,
279 bool TryGetNodeAttr(const AttrSlice& attrs, StringPiece attr_name,
281 bool TryGetNodeAttr(const AttrSlice& attrs, StringPiece attr_name,
283 bool TryGetNodeAttr(const AttrSlice& attrs, StringPiece attr_name,
[all …]
Dmemory_types.cc158 if (TryGetNodeAttr(ndef, "_input_hostmem", &hostmem_attr)) { in MemoryTypesForNode()
166 if (TryGetNodeAttr(ndef, "_output_hostmem", &hostmem_attr)) { in MemoryTypesForNode()
Dnode_def_util.cc233 bool TryGetNodeAttr(const AttrSlice& attrs, StringPiece attr_name, \
248 bool TryGetNodeAttr(const AttrSlice& attrs, StringPiece attr_name, \
344 bool TryGetNodeAttr(const AttrSlice& attrs, StringPiece attr_name, in TryGetNodeAttr() function
361 bool TryGetNodeAttr(const AttrSlice& attrs, StringPiece attr_name, in TryGetNodeAttr() function
398 bool TryGetNodeAttr(const AttrSlice& attrs, StringPiece attr_name, in TryGetNodeAttr() function
421 bool TryGetNodeAttr(const AttrSlice& attrs, StringPiece attr_name, in TryGetNodeAttr() function
Dfunction.cc637 if (!TryGetNodeAttr(n, "T", &dt)) { in Print()
1531 if (!TryGetNodeAttr(ndef, kFuncAttr, &forward_func_attrs)) { in GetAttrImpl()
1584 if (fdef && TryGetNodeAttr(AttrSlice(&fdef->attr()), attr, value)) { in GetAttr()
/external/tensorflow/tensorflow/core/common_runtime/
Dlower_functional_ops.cc44 bool found = TryGetNodeAttr(n->attrs(), attr_name, &match); in CheckBoolAttr()
51 bool found = TryGetNodeAttr(n->attrs(), attr_name, &match); in CheckStringAttr()
Dlower_function_call_op.cc39 TryGetNodeAttr(n->attrs(), kLowerAsMultiDeviceFunctionAttr, &match); in LowerAsMultiDeviceFunction()
Dmkl_layout_pass.cc1636 bool has_attr = TryGetNodeAttr(n->def(), "alpha", &alpha); in LeakyReluRewrite()
1660 TryGetNodeAttr(n->def(), "narrow_range", &narrow_range); in QuantizeOpRewrite()
1661 TryGetNodeAttr(n->def(), "axis", &axis); in QuantizeOpRewrite()
1780 if (!TryGetNodeAttr(n->def(), "T", &T) || in FusedConv2DRewrite()
1808 if (!TryGetNodeAttr(n->def(), "T", &T) || in FusedDepthwiseConv2DRewrite()
2193 if (TryGetNodeAttr(n->def(), "T", &T) && in GetNodeProducingMklTensor()
2652 if (TryGetNodeAttr(orig_node->def(), "explicit_paddings", in CopyAttrsConv()
3685 if (TryGetNodeAttr(n->def(), "Tinput", &Tinput) && in CheckForQuantizedNodeRewrite()
3686 TryGetNodeAttr(n->def(), "Tfilter", &Tfilter) && in CheckForQuantizedNodeRewrite()
3690 } else if (TryGetNodeAttr(n->def(), "T1", &T1) && in CheckForQuantizedNodeRewrite()
[all …]
Daccumulate_n_optimizer.cc88 bool found = TryGetNodeAttr(frame->attrs(), kParallelIterationsAttrName, in Run()
Dgraph_execution_state.cc477 if (!TryGetNodeAttr(node, "dtype", type) && in GetFeedShapeAndTypeFromAttribute()
478 !TryGetNodeAttr(node, "T", type)) { in GetFeedShapeAndTypeFromAttribute()
Dgraph_constructor.cc792 if (!TryGetNodeAttr(node->attrs(), kAttrName, &shape_attrs)) { in ValidateShape()
1006 if (!TryGetNodeAttr(node->attrs(), kColocationAttrName, &coloc_values)) in UpdateUniquifiedColocationNames()
Dinline_function_utils.cc288 if (TryGetNodeAttr(attr, kNoInlineAttr, &noinline) && noinline) { in ValidateNoInline()
/external/tensorflow/tensorflow/compiler/jit/
Dforce_xla_constants_on_host_pass_test.cc99 EXPECT_TRUE(TryGetNodeAttr(node->def(), "_input_hostmem", &hostmem_attr)); in TEST()
Dencapsulate_xla_computations_pass.cc46 if (!TryGetNodeAttr(n->attrs(), kXlaClusterIdAttr, &name)) continue; in IsCpuGpuCompile()
61 if (!TryGetNodeAttr(n.attrs(), "_is_guaranteed_constant", in is_guaranteed_constant()
Dmark_for_compilation_pass.cc695 if (!TryGetNodeAttr(n->def(), "dtype", &dtype) || !DataTypeIsInteger(dtype)) { in IsScalarIntegerResourceOperation()
712 if (!TryGetNodeAttr(const_input->def(), "value", &proto)) { in IsScalarIntegerResourceOperation()
981 if (TryGetNodeAttr(node->attrs(), attr_name, &attr_value)) { in GetNodeOrFuncAttr()
Dnode_matchers.cc140 if (!TryGetNodeAttr(node->def(), "value", &proto)) { in MatchAndExplain()
Dencapsulate_subgraphs_pass.cc1330 TryGetNodeAttr(node.attrs(), kXlaCompiledKernelAttr, &is_compiled) && in IsXlaCompiledKernel()
/external/tensorflow/tensorflow/core/kernels/
Dcwise_ops_common.cc62 TryGetNodeAttr(ctx->op_kernel().def(), "incompatible_shape_error", in BinaryOpState()
/external/tensorflow/tensorflow/core/grappler/
Dgrappler_item.cc161 if (TryGetNodeAttr(attrs, "_grappler_do_not_remove", &do_not_remove) && in NodesToPreserve()
/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Ddistributed_tpu_configuration_rewrite_pass.cc278 TryGetNodeAttr(configuration_node_def, in Run()
Dencapsulate_tpu_computations_pass.cc690 if (!TryGetNodeAttr(n->def(), kXlaIsLiftedArgAttrName, &is_lifted_arg) || in MoveHeadOutsideCompilationToHost()
691 !TryGetNodeAttr(n->def(), kOutsideCompilationAttr, in MoveHeadOutsideCompilationToHost()
2549 if (TryGetNodeAttr(n->attrs(), kTPUReplicateAttr, &name) && in BuildTPUReplicateOps()
2550 !TryGetNodeAttr(n->attrs(), kOutsideCompilationAttr, &name)) { in BuildTPUReplicateOps()
2890 if (TryGetNodeAttr(node->attrs(), kPivotForClusterAttr, &cluster_name)) { in ProcessHeadTailOutsideCompilation()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dremapper.cc515 if (!TryGetNodeAttr(*squeeze_node_def, "squeeze_dims", &dims)) return false; in IsCpuCompatibleDataType()
588 if (!TryGetNodeAttr(*node_def, "epsilon", &matched->epsilon)) return false; in IsCpuCompatibleDataType()
781 if (!TryGetNodeAttr(*node_def, kIsTraining, &is_training)) return false; in IsCpuCompatibleDataType()
1700 if (!TryGetNodeAttr(*node_def, kIsTraining, &is_training)) return false; in IsCpuCompatibleDataType()
Dfunction_optimizer.cc151 return TryGetNodeAttr(attr, kNoSpecializeAttr, &nospecialize) && nospecialize; in MarkedNoSpecialize()
799 bool found = TryGetNodeAttr(n->attrs(), attr_name, &match); in CheckBoolAttr()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_op_registry.cc400 if (TryGetNodeAttr(node_def, kXlaCompileTimeConstantInputsAttr, in CompileTimeConstantInputs()
/external/tensorflow/tensorflow/cc/framework/
Dscope.cc275 if (TryGetNodeAttr(attrs, kColocationAttrName, &node_constraints)) { in GetColocationConstraints()
/external/tensorflow/tensorflow/core/graph/
Dgraph_partition.cc959 if (!TryGetNodeAttr(*ndef, "send_device_incarnation", &incarnation) || in SetIncarnation()

12