Home
last modified time | relevance | path

Searched refs:input_properties (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/costs/
Dgraph_properties.cc309 const auto& input_properties = input_properties_map.at(node.name()); in VerboseLogUnknownDimensionSources() local
313 for (const auto& input_prop : input_properties) { in VerboseLogUnknownDimensionSources()
327 for (const auto& input_prop : input_properties) { in VerboseLogUnknownDimensionSources()
2186 auto& input_properties = input_properties_[node.name()]; in InferStatically() local
2189 CHECK_EQ(input_properties.size(), 0); in InferStatically()
2191 input_properties.resize(ic->num_inputs()); in InferStatically()
2195 &input_properties[i]); in InferStatically()
2201 *input_properties[i].mutable_value() = raw_val; in InferStatically()
2204 *input_properties[i].mutable_value() = *ctx->input_tensor_protos[i]; in InferStatically()
2209 *input_properties[i].mutable_value() = MakeTensorProtoFromShape( in InferStatically()
Dvirtual_scheduler.cc476 auto& inputs = node_state.input_properties; in MaybeUpdateInputOutput()
638 for (auto& input : node_state.input_properties) { in GetCurrNode()
663 node_state.input_properties = in GetNodeStateOrCreateIt()
Dvirtual_scheduler.h46 std::vector<OpInfo::TensorProperties> input_properties; member