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

  /external/tensorflow/tensorflow/compiler/xla/service/
hlo_proto_util.cc 61 std::vector<const ShapeProto*> parameter_shapes; local
64 parameter_shapes.push_back(&shape);
66 return parameter_shapes;
  /external/tensorflow/tensorflow/compiler/xla/tests/
params_test.cc 355 std::vector<Shape> parameter_shapes; local
362 parameter_shapes.push_back(literal.shape());
373 parameter_shapes.push_back(bool_literal.shape());
378 Shape while_shape = ShapeUtil::MakeTupleShape(parameter_shapes);

Completed in 152 milliseconds