Home
last modified time | relevance | path

Searched defs:input_properties (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2tensorrt/segment/
Dsegment.cc523 absl::Span<const OpInfo::TensorProperties> input_properties = in OperationCanBeTranslatedToImplicitBatch() local
656 const std::vector<OpInfo::TensorProperties>& input_properties = in GetClusterBatchSizeForNode() local
/external/tensorflow/tensorflow/core/grappler/costs/
Dvirtual_scheduler.h53 std::vector<OpInfo::TensorProperties> input_properties; member
Dop_level_cost_estimator.cc204 for (const OpInfo::TensorProperties& input_properties : op_info.inputs()) { in CwiseOutputElementCount() local
219 for (const OpInfo::TensorProperties& input_properties : op_info.inputs()) { in CwiseOutputElementCount() local
Dgraph_properties.cc316 const auto& input_properties = input_properties_map.at(node.name()); in VerboseLogUnknownDimensionSources() local
2626 auto& input_properties = input_properties_[node.name()]; in InferStatically() local