Home
last modified time | relevance | path

Searched defs:shape_proto (Results 1 – 19 of 19) sorted by relevance

/external/tensorflow/tensorflow/contrib/periodic_resample/ops/
Darray_ops_test.cc29 TensorShapeProto shape_proto; in TEST() local
/external/tensorflow/tensorflow/compiler/xla/
Dshape.cc24 Shape::Shape(const ShapeProto& shape_proto) { in Shape()
187 for (const ShapeProto& shape_proto : program_shape_proto.parameters()) { in ProgramShape() local
Dliteral_test.cc1448 void SetDefaultLayoutOnProto(ShapeProto* shape_proto) { in SetDefaultLayoutOnProto()
/external/tensorflow/tensorflow/compiler/jit/
Dencapsulate_util_test.cc49 TensorShapeProto shape_proto; in TEST() local
Dencapsulate_subgraphs_pass.cc1218 TensorShapeProto shape_proto; in AddHostComputeKeyPlaceholder() local
/external/tensorflow/tensorflow/tools/graph_transforms/
Dsummarize_graph_main.cc46 TensorShapeProto shape_proto = node->attr().at("shape").shape(); in PrintNodeInfo() local
83 TensorShapeProto shape_proto = node->attr().at("shape").shape(); in PrintBenchmarkUsage() local
Dfold_constants_test.cc146 auto* shape_proto = shape_attr.mutable_list()->add_shape(); in TestPreserveOutputShapes() local
/external/tensorflow/tensorflow/core/grappler/
Dgrappler_item_builder.cc265 TensorShapeProto shape_proto; in GrapplerItemFromMetaGraphDef() local
459 TensorShapeProto shape_proto; in GrapplerItemFromMetaGraphDef() local
/external/tensorflow/tensorflow/core/nccl/
Dnccl_rewrite.cc184 TensorShapeProto shape_proto; in ReplaceBroadcast() local
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dscoped_allocator_optimizer_test.cc98 TensorShapeProto shape_proto; in SetShapes() local
/external/tensorflow/tensorflow/core/ops/
Dmath_ops_test.cc219 auto shape_proto = [](std::initializer_list<int64> dim_sizes) { in TEST() local
Darray_ops_test.cc139 auto* shape_proto = tensor_proto.mutable_tensor_shape(); in TEST() local
1781 TensorShapeProto shape_proto; in TEST() local
/external/tensorflow/tensorflow/core/framework/
Dtensor_util.cc179 TensorShapeProto* shape_proto) { in SetTensorProtoShape()
Dop_kernel_test.cc737 TensorShapeProto shape_proto; member in tensorflow::__anon74460fe50111::GetAttrKernel
/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Dllvm_util.cc250 ShapeProto shape_proto; in DecodeSelfDescribingShapeConstant() local
/external/tensorflow/tensorflow/compiler/xla/service/
Dservice.cc831 for (const ShapeProto& shape_proto : arg->input_shape_with_layout()) { in Compile() local
Dhlo_instruction.cc449 for (const ShapeProto& shape_proto : in CreateFromProto() local
/external/tensorflow/tensorflow/cc/framework/
Dcc_op_gen.cc134 string PrintTensorShape(const TensorShapeProto& shape_proto) { in PrintTensorShape()
/external/tensorflow/tensorflow/core/grappler/costs/
Dgraph_properties.cc374 auto* shape_proto = tensor_proto.mutable_tensor_shape(); in MakeTensorProtoFromShape() local