Home
last modified time | relevance | path

Searched refs:ToProto (Results 1 – 25 of 113) sorted by relevance

12345

/external/tensorflow/tensorflow/compiler/mlir/xla/
Dtype_to_shape_test.cc81 TypeToShape(VectorType::get({8, 128}, b.getIntegerType(32))).ToProto(), in TEST()
83 ShapeUtil::MakeShape(PrimitiveType::S32, {8, 128}).ToProto())); in TEST()
85 TypeToShape(VectorType::get({8, 128}, b.getF32Type())).ToProto(), in TEST()
87 ShapeUtil::MakeShape(PrimitiveType::F32, {8, 128}).ToProto())); in TEST()
91 TypeToShape(VectorType::get({8, 128}, b.getIntegerType(17))).ToProto(), in TEST()
92 EqualsProto(Shape().ToProto())); in TEST()
101 TypeToShape(MemRefType::get({8, 128}, b.getF32Type())).ToProto(), in TEST()
103 ShapeUtil::MakeShape(PrimitiveType::F32, {8, 128}).ToProto())); in TEST()
105 TypeToShape(MemRefType::get({100, 13, 210}, b.getF32Type())).ToProto(), in TEST()
107 ShapeUtil::MakeShape(PrimitiveType::F32, {100, 13, 210}).ToProto())); in TEST()
[all …]
/external/perfetto/src/perfetto_cmd/
Dpbtxt_to_pb_unittest.cc48 TraceConfig ToProto(const std::string& input) { in ToProto() function
62 TraceConfig config = ToProto(R"( in TEST()
69 TraceConfig config = ToProto(R"( in TEST()
78 TraceConfig config = ToProto(R"( in TEST()
85 EXPECT_EQ(ToProto(R"( in TEST()
89 EXPECT_EQ(ToProto(R"( in TEST()
94 EXPECT_EQ(ToProto(R"( in TEST()
98 EXPECT_EQ(ToProto(R"( in TEST()
103 EXPECT_EQ(ToProto(R"( in TEST()
107 EXPECT_EQ(ToProto(R"( in TEST()
[all …]
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_instructions.cc93 HloInstructionProto HloBatchNormInstruction::ToProto() const { in ToProto() function in xla::HloBatchNormInstruction
94 HloInstructionProto proto = HloInstruction::ToProto(); in ToProto()
174 HloInstructionProto HloFftInstruction::ToProto() const { in ToProto() function in xla::HloFftInstruction
175 HloInstructionProto proto = HloInstruction::ToProto(); in ToProto()
215 HloInstructionProto HloCompareInstruction::ToProto() const { in ToProto() function in xla::HloCompareInstruction
216 HloInstructionProto proto = HloInstruction::ToProto(); in ToProto()
291 HloInstructionProto HloTriangularSolveInstruction::ToProto() const { in ToProto() function in xla::HloTriangularSolveInstruction
292 HloInstructionProto proto = HloInstruction::ToProto(); in ToProto()
333 HloInstructionProto HloCholeskyInstruction::ToProto() const { in ToProto() function in xla::HloCholeskyInstruction
334 HloInstructionProto proto = HloInstruction::ToProto(); in ToProto()
[all …]
Dhlo_instructions.h37 HloInstructionProto ToProto() const override;
112 HloInstructionProto ToProto() const override;
140 HloInstructionProto ToProto() const override;
166 HloInstructionProto ToProto() const override;
191 HloInstructionProto ToProto() const override;
226 HloInstructionProto ToProto() const override;
244 HloInstructionProto ToProto() const override;
322 HloInstructionProto ToProto() const override;
364 HloInstructionProto ToProto() const override;
400 HloInstructionProto ToProto() const override;
[all …]
Dhlo_proto_util.cc27 BufferAssignmentProto proto_assignment = assignment.ToProto(); in MakeHloProto()
34 HloModuleProto proto_module = module.ToProto(); in MakeHloProto()
Dhlo_module_group.cc59 HloModuleGroupProto HloModuleGroup::ToProto() const { in ToProto() function in xla::HloModuleGroup
63 *proto.add_hlo_modules() = module->ToProto(); in ToProto()
Dhlo_module.cc247 HloModuleProto HloModule::ToProto() const { in ToProto() function in xla::HloModule
254 HloComputationProto computation_proto = computation->ToProto(); in ToProto()
258 *proto.mutable_schedule() = schedule().ToProto().ValueOrDie(); in ToProto()
261 entry_computation_layout().ComputeProgramShape().ToProto(); in ToProto()
262 *proto.mutable_input_output_alias() = input_output_alias_config().ToProto(); in ToProto()
264 dynamic_parameter_binding().ToProto(); in ToProto()
Dhlo_module_group_test.cc57 group.ToProto(), {group.module(0).config()})); in TEST_F()
102 group.ToProto(), {group.module(0).config(), in TEST_F()
Dhlo_module_test.cc220 HloModule::CreateFromProto(module->ToProto(), module->config())); in TEST_F()
241 HloModule::CreateFromProto(module->ToProto(), module->config())); in TEST_F()
307 HloModule::CreateFromProto(module->ToProto(), module->config())); in TEST_F()
/external/tensorflow/tensorflow/compiler/xrt/tests/
Draw_api_test.cc81 return tuple.ToProto(); in TwoElementTuple()
89 return tuple.ToProto(); in BasedTwoElementTuple()
94 return scalar.ToProto(); in ScalarLiteral()
103 return nested.ToProto(); in NestedTuple()
113 return nested1.ToProto(); in MakeTuple0()
118 return array.ToProto(); in FloatVector()
125 return array.ToProto(); in FloatMatrix()
153 << a.ToProto().DebugString() << "\n!=\n" in CompareLiteralToLiteralProto()
163 << a.ToProto().DebugString() << "\n!=\n" in CompareLiterals()
164 << b.ToProto().DebugString(); in CompareLiterals()
[all …]
/external/tensorflow/tensorflow/compiler/xla/tests/
Dreplay_test.cc58 ASSERT_TRUE(protobuf_util::ProtobufEquals(original_shape->ToProto(), in TEST_F()
59 replayed_shape->ToProto())); in TEST_F()
91 ASSERT_TRUE(protobuf_util::ProtobufEquals(original_shape->ToProto(), in XLA_TEST_F()
92 replayed_shape->ToProto())); in XLA_TEST_F()
138 ASSERT_TRUE(protobuf_util::ProtobufEquals(original_shape->ToProto(), in TEST_F()
139 replayed_shape->ToProto())); in TEST_F()
Dclient_test.cc54 .ToProto(); in XLA_TEST_F()
89 .ToProto(); in XLA_TEST_F()
/external/tensorflow/tensorflow/core/distributed_runtime/
Dmessage_wrappers_test.cc176 CheckRunStepRequest(ProtoRunStepRequest(&in_memory_request.ToProto())); in TEST()
177 CheckRunStepRequest(ProtoRunStepRequest(&proto_request.ToProto())); in TEST()
188 &mutable_proto_run_step_request.ToProto()); in TEST()
195 CheckRunGraphRequest(ProtoRunGraphRequest(&request.ToProto())); in TEST()
203 CheckRunGraphRequest(ProtoRunGraphRequest(&request.ToProto())); in TEST()
211 CheckRunGraphRequest(ProtoRunGraphRequest(&request.ToProto())); in TEST()
219 CheckRunGraphRequest(ProtoRunGraphRequest(&request.ToProto())); in TEST()
227 CheckRunGraphRequest(ProtoRunGraphRequest(&request.ToProto())); in TEST()
235 CheckRunGraphRequest(ProtoRunGraphRequest(&request.ToProto())); in TEST()
Dmessage_wrappers.h99 virtual const RunStepRequest& ToProto() const = 0;
133 const RunStepRequest& ToProto() const override;
184 const RunStepRequest& ToProto() const override;
225 const RunStepRequest& ToProto() const override;
301 virtual const RunGraphRequest& ToProto() const = 0;
346 const RunGraphRequest& ToProto() const override;
408 const RunGraphRequest& ToProto() const override;
451 const RunGraphRequest& ToProto() const override;
Dmessage_wrappers.cc112 return ToProto().DebugString(); in DebugString()
115 const RunStepRequest& InMemoryRunStepRequest::ToProto() const { in ToProto() function in tensorflow::InMemoryRunStepRequest
227 const RunStepRequest& MutableProtoRunStepRequest::ToProto() const { in ToProto() function in tensorflow::MutableProtoRunStepRequest
291 const RunStepRequest& ProtoRunStepRequest::ToProto() const { return *request_; } in ToProto() function in tensorflow::ProtoRunStepRequest
403 const RunGraphRequest& InMemoryRunGraphRequest::ToProto() const { in ToProto() function in tensorflow::InMemoryRunGraphRequest
554 const RunGraphRequest& MutableProtoRunGraphRequest::ToProto() const { in ToProto() function in tensorflow::MutableProtoRunGraphRequest
615 const RunGraphRequest& ProtoRunGraphRequest::ToProto() const { in ToProto() function in tensorflow::ProtoRunGraphRequest
/external/tensorflow/tensorflow/core/kernels/
Dgpu_utils.cc141 *instr.mutable_input() = input_desc.ToProto(element_type); in LogConvAutotuneResults()
142 *instr.mutable_filter() = filter_desc.ToProto(element_type); in LogConvAutotuneResults()
143 *instr.mutable_output() = output_desc.ToProto(element_type); in LogConvAutotuneResults()
144 *instr.mutable_conv_desc() = conv_desc.ToProto(); in LogConvAutotuneResults()
183 *instr.mutable_input() = input_desc.ToProto(element_type); in LogFusedConvForwardAutotuneResults()
184 *instr.mutable_filter() = filter_desc.ToProto(element_type); in LogFusedConvForwardAutotuneResults()
185 *instr.mutable_output() = output_desc.ToProto(element_type); in LogFusedConvForwardAutotuneResults()
186 *instr.mutable_conv_desc() = conv_desc.ToProto(); in LogFusedConvForwardAutotuneResults()
/external/tensorflow/tensorflow/compiler/xla/
Dshape.h43 ShapeProto ToProto() const;
159 string SerializeAsString() const { return ToProto().SerializeAsString(); } in SerializeAsString()
160 string ShortDebugString() const { return ToProto().ShortDebugString(); } in ShortDebugString()
161 string DebugString() const { return ToProto().DebugString(); } in DebugString()
263 ProgramShapeProto ToProto() const;
313 string ShortDebugString() const { return ToProto().ShortDebugString(); } in ShortDebugString()
314 string DebugString() const { return ToProto().DebugString(); } in DebugString()
Dshape.cc58 ShapeProto Shape::ToProto() const { in ToProto() function in xla::Shape
70 *proto.add_tuple_shapes() = shape.ToProto(); in ToProto()
73 *proto.mutable_layout() = layout().ToProto(); in ToProto()
199 ProgramShapeProto ProgramShape::ToProto() const { in ToProto() function in xla::ProgramShape
202 *proto.add_parameters() = shape.ToProto(); in ToProto()
204 *proto.mutable_result() = result().ToProto(); in ToProto()
Dlayout.cc24 TileProto Tile::ToProto() const { in ToProto() function in xla::Tile
63 LayoutProto Layout::ToProto() const { in ToProto() function in xla::Layout
71 *proto.add_tiles() = tile.ToProto(); in ToProto()
Dliteral_test.cc1454 LiteralProto p = m.ToProto(); in TEST_F()
1786 return Literal::CreateFromProto(literal.ToProto()).ValueOrDie(); in TEST_F()
1807 *proto.mutable_shape() = ShapeUtil::MakeShape(F32, {3}).ToProto(); in TEST_F()
1817 *proto.mutable_shape() = ShapeUtil::MakeShape(F32, {3}).ToProto(); in TEST_F()
1826 LiteralProto proto = literal.ToProto(); in TEST_F()
1852 *proto.mutable_shape() = ShapeUtil::MakeShape(F32, {3}).ToProto(); in TEST_F()
1865 *proto.mutable_shape() = ShapeUtil::MakeShape(F32, {42, 2}).ToProto(); in TEST_F()
1878 *proto.mutable_shape() = ShapeUtil::MakeShape(S32, {2}).ToProto(); in TEST_F()
1891 *proto.mutable_shape() = ShapeUtil::MakeShape(PRED, {2, 2}).ToProto(); in TEST_F()
1908 .ToProto(); in TEST_F()
[all …]
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.cc315 *instr.mutable_shape() = param_shape.ToProto(); in SetDynamicBinding()
398 instruction_shapes_[index]->ToProto(); in Build()
405 *entry.mutable_program_shape() = program_shape.ToProto(); in Build()
431 dynamic_parameter_binding_.ToProto(); in Build()
468 *module->mutable_input_output_alias() = config.ToProto(); in PopulateInputOutputAlias()
478 *instr.mutable_shape() = shape.ToProto(); in InDimBroadcast()
542 *instr.mutable_shape() = shape.ToProto(); in UnaryOp()
557 *instr.mutable_shape() = shape.ToProto(); in BinaryOp()
675 *instr.mutable_shape() = status_or_shape.ConsumeValueOrDie().ToProto(); in TernaryOp()
687 *instr.mutable_shape() = scalar.shape().ToProto(); in ConstantLiteral()
[all …]
Dsharding_builder.cc39 *result.mutable_tile_shape() = tile_shape.ToProto(); in Tile()
55 *result.mutable_tile_shape() = tile_shape.ToProto(); in Tile1D()
Dclient.cc46 *request.mutable_shape_with_layout() = shape_with_layout->ToProto(); in Transfer()
71 *request.mutable_literal() = literal.ToProto(); in TransferToServer()
99 *request.mutable_literal() = literal.ToProto(); in TransferToInfeed()
127 *request.mutable_shape_with_layout() = shape_with_layout->ToProto(); in TransferFromOutfeed()
189 *request.mutable_output_layout() = output_layout->ToProto(); in ComputeConstant()
236 *request.add_input_shape_with_layout() = arg_shape.ToProto(); in Compile()
/external/tensorflow/tensorflow/core/framework/
Dvariant_tensor_data.cc48 void VariantTensorData::ToProto(VariantTensorDataProto* proto) const { in ToProto() function in tensorflow::VariantTensorData
82 ToProto(&proto); in SerializeAsString()
88 ToProto(&proto); in SerializeToString()
Dfunction_test.cc1160 EXPECT_EQ(0, lib_def.ToProto().function_size()); in TEST()
1161 EXPECT_EQ(0, lib_def.ToProto().gradient_size()); in TEST()
1175 EXPECT_EQ(1, lib_def.ToProto().function_size()); in TEST()
1176 EXPECT_EQ(1, lib_def.ToProto().gradient_size()); in TEST()
1196 EXPECT_EQ(1, lib_def.ToProto().function_size()); in TEST()
1197 EXPECT_EQ(1, lib_def.ToProto().gradient_size()); in TEST()
1211 EXPECT_EQ(1, lib_def.ToProto().function_size()); in TEST()
1212 EXPECT_EQ(1, lib_def.ToProto().gradient_size()); in TEST()
1232 EXPECT_EQ(1, lib_def.ToProto().function_size()); in TEST()
1233 EXPECT_EQ(1, lib_def.ToProto().gradient_size()); in TEST()
[all …]

12345