HomeSort by relevance Sort by last modified time
    Searched refs:GetInputProperties (Results 1 - 14 of 14) sorted by null

  /external/tensorflow/tensorflow/core/grappler/optimizers/
shape_optimizer.cc 71 const auto& input_props = properties.GetInputProperties(node.name());
105 const auto& prop1 = properties.GetInputProperties(input1.node->name());
106 const auto& prop2 = properties.GetInputProperties(input2.node->name());
constant_folding.cc 148 properties.GetInputProperties(node->name()).empty()) {
151 const auto& axis_input = properties.GetInputProperties(node->name()).back();
361 properties.GetInputProperties(node->name());
512 properties.GetInputProperties(shape_node.name());
682 properties.GetInputProperties(node->name());
    [all...]
static_schedule.cc 38 properties.GetInputProperties(node.name());
arithmetic_optimizer.cc     [all...]
remapper.cc 199 ctx.graph_properties.GetInputProperties(matched.conv2d->name());
437 const auto& props = ctx.graph_properties.GetInputProperties(node->name());
    [all...]
memory_optimizer.cc     [all...]
layout_optimizer.cc 645 graph_properties_.GetInputProperties(node_->name())[pos].dtype();
    [all...]
  /external/tensorflow/tensorflow/core/grappler/costs/
graph_properties_test.cc 135 EXPECT_EQ(1, properties.GetInputProperties(node.name()).size());
146 const auto in_props = properties.GetInputProperties(node.name());
178 EXPECT_EQ(1, properties.GetInputProperties(node.name()).size());
184 const auto in_props = properties.GetInputProperties(node.name());
187 const auto cleared_props = properties.GetInputProperties(node.name());
207 EXPECT_EQ(0, properties.GetInputProperties(node.name()).size());
213 const auto props = properties.GetInputProperties(node.name());
219 const auto props = properties.GetInputProperties(node.name());
539 const auto props = properties.GetInputProperties("Less");
    [all...]
graph_properties.h 116 const std::vector<OpInfo::TensorProperties>& GetInputProperties(
graph_memory.cc 91 properties->GetInputProperties(node.name());
virtual_scheduler.cc 664 graph_properties_->GetInputProperties(node->name());
    [all...]
graph_properties.cc     [all...]
  /external/tensorflow/tensorflow/python/grappler/
model_analyzer.cc 90 properties.GetInputProperties(node->name());
  /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
convert_nodes.cc 247 void GetInputProperties(const grappler::GraphProperties& graph_properties,
251 auto input_params = graph_properties.GetInputProperties(node->name());
    [all...]

Completed in 482 milliseconds