moco::StridedSliceGraphBuilder Class Reference

#include <StridedSlice.h>

Collaboration diagram for moco::StridedSliceGraphBuilder:

Public Member Functions
bool	validate (const tensorflow::NodeDef &) const final
void	build (const tensorflow::NodeDef &, GraphBuilderContext *) const final
Public Member Functions inherited from moco::GraphBuilder
virtual	~GraphBuilder ()

Detailed Description

Definition at line 25 of file StridedSlice.h.

Member Function Documentation

build()

void moco::StridedSliceGraphBuilder::build ( const tensorflow::NodeDef &  node, GraphBuilderContext *  context  ) const

finalvirtual

Implements moco::GraphBuilder.

Definition at line 139 of file StridedSlice.cpp.

{
  assert(context != nullptr);
 
  loco::Graph *graph = context->graph();
  SymbolTable *tensor_names = context->tensor_names();
  UpdateQueue *updates = context->updates();
 
  std::string node_name = node.name();
 
  auto stridedslice = graph->nodes()->create<TFStridedSlice>();
  stridedslice->name(node_name);
 
  // read attributes
  auto begin_mask = plier::tf::get_int_attr(node, "begin_mask");
  auto end_mask = plier::tf::get_int_attr(node, "end_mask");
  auto ellipsis_mask = plier::tf::get_int_attr(node, "ellipsis_mask");
  auto new_axis_mask = plier::tf::get_int_attr(node, "new_axis_mask");
  auto shrink_axis_mask = plier::tf::get_int_attr(node, "shrink_axis_mask");
 
  stridedslice->begin_mask(begin_mask);
  stridedslice->end_mask(end_mask);
  stridedslice->ellipsis_mask(ellipsis_mask);
  stridedslice->new_axis_mask(new_axis_mask);
  stridedslice->shrink_axis_mask(shrink_axis_mask);
 
  // TODO support general mask values: we support only this limited case for now
  assert(begin_mask == 0);
  assert(end_mask == 0);
  assert(ellipsis_mask == 0);
  assert(new_axis_mask == 0);
  assert(shrink_axis_mask == 1);
 
  // save the name for graph link updates
  TensorName output_name(node_name, 0);
  tensor_names->enroll(output_name, stridedslice);
 
  std::vector<TensorName> input_names;
  input_names.push_back(TensorName(node.input(0))); // input
  input_names.push_back(TensorName(node.input(1))); // begin
  input_names.push_back(TensorName(node.input(2))); // end
  input_names.push_back(TensorName(node.input(3))); // strides
 
  auto tfconv2d_update = std::make_unique<TFStridedSliceGraphUpdate>(stridedslice, input_names);
 
  updates->enroll(std::move(tfconv2d_update));
}

References moco::SymbolTable::enroll(), moco::UpdateQueue::enroll(), plier::tf::get_int_attr(), moco::GraphBuilderContext::graph(), moco::TFNode::name(), moco::GraphBuilderContext::tensor_names(), and moco::GraphBuilderContext::updates().

validate()

bool moco::StridedSliceGraphBuilder::validate ( const tensorflow::NodeDef &  node ) const

finalvirtual

Implements moco::GraphBuilder.

Definition at line 126 of file StridedSlice.cpp.

{
  // TODO support node.input_size() == 3 where strides is None
  if (node.input_size() != 4)
    return false;
 
  if (!plier::tf::has_attrs(node, {"T", "Index", "begin_mask", "end_mask", "ellipsis_mask",
                                   "new_axis_mask", "shrink_axis_mask"}))
    return false;
 
  return true;
}

References plier::tf::has_attrs().

---

The documentation for this class was generated from the following files:

• compiler/moco/import/include/moco/Import/Nodes/StridedSlice.h
• compiler/moco/import/src/Nodes/StridedSlice.cpp