Home
last modified time | relevance | path

Searched refs:GetTensorProperties (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/optimizers/
Dgraph_optimizer_stage_test.cc171 TF_CHECK_OK(stage.GetTensorProperties("Add", &add_properties)); in TEST_F()
175 TF_CHECK_OK(stage.GetTensorProperties("a:0", &a_properties)); in TEST_F()
179 TF_CHECK_OK(stage.GetTensorProperties("b:0", &b_properties)); in TEST_F()
Dgraph_optimizer_stage.h71 Status GetTensorProperties(const GraphOptimizerContext& ctx,
192 Status GetTensorProperties(const string& tensor, in GetTensorProperties() function
194 return ::tensorflow::grappler::GetTensorProperties(ctx_, tensor, in GetTensorProperties()
Darithmetic_optimizer.cc349 TF_RETURN_IF_ERROR(GetTensorProperties(*input_tensor, &properties)); in AbsorbInputByOptimizedNodesGroup()
361 GetTensorProperties(root_node->name(), &root_node_output_properties)); in CreateOptimizedNodesGroup()
383 Status has_input_properties = GetTensorProperties(input, &input_props); in HasAllInputsBroadcastableToShape()
470 Status has_properties = GetTensorProperties(node->name(), &properties); in IsSupported()
492 Status has_properties = GetTensorProperties(node.name(), &properties); in IsAbsorbableByOptimizedNodesGroup()
797 TF_RETURN_IF_ERROR(GetTensorProperties(input->input(0), &properties0)); in GetCommonFactors()
798 TF_RETURN_IF_ERROR(GetTensorProperties(input->input(1), &properties1)); in GetCommonFactors()
860 TF_RETURN_IF_ERROR(GetTensorProperties(unique_factors->front(), &lhs)); in GetUniqueFactors()
861 TF_RETURN_IF_ERROR(GetTensorProperties(unique_factors->back(), &rhs)); in GetUniqueFactors()
908 Status has_properties = GetTensorProperties(node->name(), &properties); in IsSupported()
[all …]
Dgraph_optimizer_stage.cc43 Status GetTensorProperties(const GraphOptimizerContext& ctx, in GetTensorProperties() function