| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | side_effect_util.cc | 46 node->ClearAttr("device_ordinal"); in SetDeviceOrdinalAttributeForNode()
56 node->ClearAttr(attr_name); in SetDeviceOrdinalAttributeForNode()
66 node->ClearAttr(attr_name); in SetDeviceOrdinalAttributeForNode()
71 node->ClearAttr("_device_ordinal"); in SetDeviceOrdinalAttributeForNode()
|
| D | rearrange_function_argument.cc | 168 n->ClearAttr("index"); in RearrangeArgNodes()
270 n->ClearAttr("index"); in RearrangeRetvalNodes()
298 n->ClearAttr("T"); in MaybeRewriteWhileNode()
356 n->ClearAttr("index"); in MaybeRewriteWhileNode()
374 n->ClearAttr(attr_name); in MaybeRewriteWhileNode()
427 n->ClearAttr("Tin"); in MaybeRewriteIfNode()
468 n->ClearAttr(attr_name); in MaybeRewriteIfNode()
502 n->ClearAttr("Tout"); in MaybeRewriteIfNode()
|
| D | tf2xla_util.cc | 207 n->ClearAttr(attr_name); in PropagateConstIntoFuncAttr()
677 node->ClearAttr(associated_function.attr_name()); in RewriteAssociatedFunction()
926 bwd_while->ClearAttr("body"); in RewriteTensorListWithConstElement()
|
| D | functionalize_control_flow.cc | 371 n->ClearAttr(func_attr); in Run()
|
| D | xla_compiler.cc | 767 fbody->arg_nodes[i]->ClearAttr("_output_shapes"); in CompileFunction()
773 fbody->arg_nodes[i]->ClearAttr("_output_shapes"); in CompileFunction()
|
| D | functionalize_cond.cc | 1591 n->ClearAttr(kCondGroupDebugAttr); in DumpGraphWithCondState()
|
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | encapsulate_tpu_computations_pass.cc | 345 xla_node->ClearAttr(TPUREPLICATE_MIRRORED_VAR_INDICES_ATTR); in UpdateMirroredVariableIndices()
568 xla_node->ClearAttr("Tinputs"); in MoveHeadOutsideCompilationToHost()
578 xla_node->ClearAttr("_variable_start_index"); in MoveHeadOutsideCompilationToHost()
584 xla_node->ClearAttr("_guaranteed_const_start_index"); in MoveHeadOutsideCompilationToHost()
655 n->ClearAttr("index"); in MoveHeadOutsideCompilationToHost()
840 n->ClearAttr("index"); in RemoveUnusedXlaInput()
911 xla_node->ClearAttr("Tinputs"); in RemoveUnusedXlaInput()
916 xla_node->ClearAttr("num_distributed_variables"); in RemoveUnusedXlaInput()
932 xla_node->ClearAttr(TPUREPLICATE_MIRRORED_VAR_INDICES_ATTR); in RemoveUnusedXlaInput()
945 xla_node->ClearAttr("Tbroadcast_inputs"); in RemoveUnusedXlaInput()
[all …]
|
| D | host_training_loop_optimization_util.cc | 471 compile_node->ClearAttr("metadata"); in AddReshardOp()
|
| D | distributed_tpu_rewrite_pass.cc | 3786 n->ClearAttr(LowerFunctionalOpsPass::kLowerAsMultiDeviceFunctionAttr); in LowerOutsideCompilationFunctionalNodes()
3788 n->ClearAttr(LowerFunctionalOpsPass::kLowerAsMultiDeviceFunctionAttr); in LowerOutsideCompilationFunctionalNodes()
3862 n->ClearAttr(kTPUReplicateAttr); in LowerOutsideCompilationFunctionalNodes()
3863 n->ClearAttr(kOutsideCompilationAttr); in LowerOutsideCompilationFunctionalNodes()
4202 node->ClearAttr(kTPUReplicateAttr); in UpdateHeadTailOutsideCompilation()
4203 node->ClearAttr(kOutsideCompilationAttr); in UpdateHeadTailOutsideCompilation()
|
| /external/tensorflow/tensorflow/c/ |
| D | python_api.cc | 44 void ClearAttr(TF_Graph* graph, TF_Operation* op, const char* attr_name, in ClearAttr() function
48 op->node.ClearAttr(attr_name); in ClearAttr()
|
| D | python_api.h | 37 void ClearAttr(TF_Graph* graph, TF_Operation* op, const char* attr_name,
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | encapsulate_util.cc | 58 n->ClearAttr(attr_name); in AppendToListAttr()
67 n->ClearAttr(attr_name); in ReplaceAttr()
292 n->ClearAttr(kXlaControlDependenciesWithinXlaClusterAttrName); in PostprocessControlEdgesBetweenOutsideCompilations()
|
| D | extract_outside_compilation_pass.cc | 407 n->ClearAttr("device_ordinal"); in ResetDeviceOrdinalToPlaceholderValue()
415 n->ClearAttr(attr_name); in ResetDeviceOrdinalToPlaceholderValue()
423 n->ClearAttr(attr_name); in ResetDeviceOrdinalToPlaceholderValue()
428 n->ClearAttr("_device_ordinal"); in ResetDeviceOrdinalToPlaceholderValue()
493 n->ClearAttr(type_attr_name); in UpdateTypesAttribute()
577 callsite_node->ClearAttr(func_attr_name); in AddFunctionWithNewName()
772 n->ClearAttr("Tin"); in PostprocessLiftedArgsForIf()
1542 while_node->ClearAttr("cond"); in AddSendLoopPredToLoopCond()
1935 n->ClearAttr("then_branch"); in ExtractOutsideCompilationForIfNode()
1940 n->ClearAttr("else_branch"); in ExtractOutsideCompilationForIfNode()
[all …]
|
| D | xla_cluster_util.cc | 220 void RemoveFromXlaCluster(Node* node) { node->ClearAttr(kXlaClusterAttr); } in RemoveFromXlaCluster()
|
| D | build_xla_ops_pass.cc | 558 n->ClearAttr(kXlaCompiledKernelAttr); in ReplaceNodeWithXlaCompileAndXlaRun()
|
| D | encapsulate_subgraphs_pass.cc | 700 image->ClearAttr(group_attribute_); in CopySubgraphNodes()
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | collective_order.cc | 172 pair.first->ClearAttr("wait_for"); in InsertControlDependencies()
|
| D | graph.h | 216 void ClearAttr(const std::string& name);
|
| D | graph.cc | 219 void Node::ClearAttr(const string& name) { in ClearAttr() function in tensorflow::Node
|
| /external/tensorflow/tensorflow/tools/def_file_filter/ |
| D | symbols_pybind.txt | 227 tensorflow::ClearAttr
|
| /external/tensorflow/tensorflow/python/client/ |
| D | tf_session_wrapper.cc | 699 tensorflow::ClearAttr(graph, op, attr_name, status.get()); in PYBIND11_MODULE()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | graph_constructor.cc | 831 node->ClearAttr(kAttrName); in ValidateShape()
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | ops.py | 2530 pywrap_tf_session.ClearAttr(self._graph._c_graph, self._c_op, attr_name)
|