Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/profiler/utils/
Dcost_utils.cc50 static OpInfo::TensorProperties GetTensorProperties(absl::string_view info) { in GetTensorProperties() function
113 *op_context.op_info.add_inputs() = GetTensorProperties(tensor); in Predict()
/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( in GetTensorProperties() function
194 return ::tensorflow::grappler::GetTensorProperties(ctx_, tensor, in GetTensorProperties()
Dgraph_optimizer_stage.cc43 Status GetTensorProperties(const GraphOptimizerContext& ctx, in GetTensorProperties() function
Darithmetic_optimizer.cc374 TF_RETURN_IF_ERROR(GetTensorProperties(*input_tensor, &properties)); in AbsorbInputByOptimizedNodesGroup()
386 GetTensorProperties(root_node->name(), &root_node_output_properties)); in CreateOptimizedNodesGroup()
408 Status has_input_properties = GetTensorProperties(input, &input_props); in HasAllInputsBroadcastableToShape()
495 Status has_properties = GetTensorProperties(node->name(), &properties); in IsSupported()
517 Status has_properties = GetTensorProperties(node.name(), &properties); in IsAbsorbableByOptimizedNodesGroup()
824 TF_RETURN_IF_ERROR(GetTensorProperties(input->input(0), &properties0)); in GetCommonFactors()
825 TF_RETURN_IF_ERROR(GetTensorProperties(input->input(1), &properties1)); in GetCommonFactors()
887 TF_RETURN_IF_ERROR(GetTensorProperties(unique_factors->front(), &lhs)); in GetUniqueFactors()
888 TF_RETURN_IF_ERROR(GetTensorProperties(unique_factors->back(), &rhs)); in GetUniqueFactors()
936 Status has_properties = GetTensorProperties(node->name(), &properties); in IsSupported()
[all …]