Home
last modified time | relevance | path

Searched refs:HumanStringWithLayout (Results 1 – 25 of 53) sorted by relevance

123

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/
H A Dlayout_util_test.cc319 EXPECT_EQ(ShapeUtil::HumanStringWithLayout(shape), "f32[2,3,4]{0,1,2}"); in TEST_F()
325 EXPECT_EQ(ShapeUtil::HumanStringWithLayout(shape), in TEST_F()
332 EXPECT_EQ(ShapeUtil::HumanStringWithLayout(shape), in TEST_F()
343 EXPECT_EQ(ShapeUtil::HumanStringWithLayout(shape), in TEST_F()
351 EXPECT_EQ(ShapeUtil::HumanStringWithLayout(shape), in TEST_F()
360 EXPECT_EQ(ShapeUtil::HumanStringWithLayout(shape), in TEST_F()
366 EXPECT_EQ(ShapeUtil::HumanStringWithLayout(shape), in TEST_F()
375 EXPECT_EQ(ShapeUtil::HumanStringWithLayout(shape), in TEST_F()
385 EXPECT_EQ(ShapeUtil::HumanStringWithLayout(shape), in TEST_F()
H A Dshape_util.cc674 /* static */ std::string ShapeUtil::HumanStringWithLayout(const Shape& shape) { in HumanStringWithLayout() function in xla::ShapeUtil
686 StrAppend(&text, HumanStringWithLayout(elem_shape)); in HumanStringWithLayout()
1188 << "shape=" << HumanStringWithLayout(shape) in PermuteDimensions()
1189 << ", new_shape=" << HumanStringWithLayout(new_shape) in PermuteDimensions()
1582 << ShapeUtil::HumanStringWithLayout(input_shape) in AlignLayouts()
1584 << ShapeUtil::HumanStringWithLayout(output_shape_with_layout); in AlignLayouts()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/tpu/
H A Dtpu_executable_interface.cc62 << " shape = " << ShapeUtil::HumanStringWithLayout(shape); in AllocateOutputMemoryWithInputReuse()
97 << " shape = " << ShapeUtil::HumanStringWithLayout(shape); in AllocateOutputMemoryWithInputReuse()
100 << ShapeUtil::HumanStringWithLayout(shape) << " -> " in AllocateOutputMemoryWithInputReuse()
101 << ShapeUtil::HumanStringWithLayout(device_shape); in AllocateOutputMemoryWithInputReuse()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xrt/
H A Dxrt_state.cc90 << xla::ShapeUtil::HumanStringWithLayout(shape) << " device_shape=" in AllocateScopedShapedBuffer()
91 << xla::ShapeUtil::HumanStringWithLayout(on_device_shape); in AllocateScopedShapedBuffer()
263 xla::ShapeUtil::HumanStringWithLayout(literal.shape()), in WriteLiteral()
264 " device=", xla::ShapeUtil::HumanStringWithLayout(on_host_shape())); in WriteLiteral()
629 xla::ShapeUtil::HumanStringWithLayout(source_subshape), in AliasBufferFrom()
630 " dest=", xla::ShapeUtil::HumanStringWithLayout(dest_subshape)); in AliasBufferFrom()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dinfeed_thunk.cc50 << ShapeUtil::HumanStringWithLayout(source_shape) in ExecuteOnStream()
52 << ShapeUtil::HumanStringWithLayout(dest_slices_[index].shape); in ExecuteOnStream()
H A Doutfeed_manager.cc51 CHECK(shape.IsArray()) << ShapeUtil::HumanStringWithLayout(shape); in TransferLiteralFromOutfeed()
67 CHECK(shape.IsArray()) << ShapeUtil::HumanStringWithLayout(shape); in TransferLiteralFromOutfeed()
H A Doutfeed_thunk.cc79 << ShapeUtil::HumanStringWithLayout(output_shape) in ExecuteOnStream()
81 << ShapeUtil::HumanStringWithLayout(source_slices_[index].shape); in ExecuteOnStream()
H A Dstream_executor_util.cc212 ShapeUtil::HumanStringWithLayout(input), in XlaConvShapesToStreamExecutorLayouts()
236 ShapeUtil::HumanStringWithLayout(filter), in XlaConvShapesToStreamExecutorLayouts()
260 ShapeUtil::HumanStringWithLayout(output), in XlaConvShapesToStreamExecutorLayouts()
H A Dfft_thunk.cc101 VLOG(3) << "Input shape: " << ShapeUtil::HumanStringWithLayout(input_shape); in RunFft()
102 VLOG(3) << "Output shape: " << ShapeUtil::HumanStringWithLayout(output_shape); in RunFft()
H A Dgpu_conv_runner.cc342 << ShapeUtil::HumanStringWithLayout(config.input_shape); in GetGpuConvConfig()
344 << ShapeUtil::HumanStringWithLayout(config.filter_shape); in GetGpuConvConfig()
346 << ShapeUtil::HumanStringWithLayout(config.output_shape); in GetGpuConvConfig()
H A Dinfeed_manager.cc70 CHECK(sub_shape.IsArray()) << ShapeUtil::HumanStringWithLayout(sub_shape); in TransferLiteralToInfeed()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tools/
H A Dinteractive_graphviz.cc305 << ShapeUtil::HumanStringWithLayout(param->shape()) << ")" in DoInfoCommand()
310 << ShapeUtil::HumanStringWithLayout(root->shape()) << ")" in DoInfoCommand()
338 std::cout << " Shape: " << ShapeUtil::HumanStringWithLayout(instr->shape()) in DoInfoCommand()
362 << ShapeUtil::HumanStringWithLayout(operand->shape()) << ")" in DoInfoCommand()
/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/kernels/
H A Doutfeed_ops.h60 << xla::ShapeUtil::HumanStringWithLayout(xla_shape_); in DoWork()
107 << xla::ShapeUtil::HumanStringWithLayout(tuple_shape_); in DoWork()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dbitcast_decomposer.cc125 << "from=" << ShapeUtil::HumanStringWithLayout(input->shape()) in Run()
126 << " to=" << ShapeUtil::HumanStringWithLayout(shape); in Run()
H A Dbitcast_decomposer_test.cc141 absl::Substitute(kModuleTemplate, ShapeUtil::HumanStringWithLayout(src), in TEST_P()
142 ShapeUtil::HumanStringWithLayout(dst)); in TEST_P()
H A Dshaped_buffer.cc92 ShapeUtil::HumanStringWithLayout(on_device_shape()), in ToString()
101 shape_str = ShapeUtil::HumanStringWithLayout(subshape); in ToString()
H A Dhlo_matchers.cc184 << ShapeUtil::HumanStringWithLayout(shape_) << ")"; in MatchAndExplain()
189 *os << ShapeUtil::HumanStringWithLayout(shape_); in DescribeTo()
H A Dhlo_input_output_alias_config.cc213 ShapeUtil::HumanStringWithLayout(param_subshape), in Verify()
215 ShapeUtil::HumanStringWithLayout(output_subshape), in Verify()
H A Dhlo_module.cc409 << ShapeUtil::HumanStringWithLayout( in CreateFromProto()
411 << ", actual: " << ShapeUtil::HumanStringWithLayout(parameter_shape); in CreateFromProto()
419 << ShapeUtil::HumanStringWithLayout(expected_program_shape.result()) in CreateFromProto()
420 << ", actual: " << ShapeUtil::HumanStringWithLayout(result_shape); in CreateFromProto()
H A Dlayout_assignment.cc258 << ShapeUtil::HumanStringWithLayout(shape_with_layout) in SetOperandLayout()
345 VLOG(3) << " : " << ShapeUtil::HumanStringWithLayout(shape_with_layout) in SetResultLayout()
400 << ShapeUtil::HumanStringWithLayout(shape_with_layout) in SetInstructionLayout()
408 ShapeUtil::HumanStringWithLayout(shape_with_layout)); in SetInstructionLayout()
954 ShapeUtil::HumanStringWithLayout(constant->literal().shape())); in CheckConstantLayout()
1163 ShapeUtil::HumanStringWithLayout(instruction_subshape), in CheckLayouts()
1164 ShapeUtil::HumanStringWithLayout(buffer->shape())); in CheckLayouts()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
H A Dir_array.cc115 << "Shape " << ShapeUtil::HumanStringWithLayout(shape) in Index()
136 << "Shape " << ShapeUtil::HumanStringWithLayout(shape) in Index()
156 << "Shape " << ShapeUtil::HumanStringWithLayout(shape) in Index()
180 << "Shape " << ShapeUtil::HumanStringWithLayout(shape) in Index()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/interpreter/
H A Dexecutable_base.cc100 ShapeUtil::HumanStringWithLayout(expected_shape), in ExecuteAsyncOnStream()
101 ShapeUtil::HumanStringWithLayout(actual_shape)); in ExecuteAsyncOnStream()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/
H A Dshape_util.cc202 << "] = " << xla::ShapeUtil::HumanStringWithLayout(shapes.back()); in GetShapeWithLayout()
222 << xla::ShapeUtil::HumanStringWithLayout(*output_shape); in GetShapeWithLayout()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/
H A Dclient_library_test_base.cc205 ShapeUtil::HumanStringWithLayout(layout))); in ComputeAndCompareLiteralWithAllOutputLayouts()
230 ShapeUtil::HumanStringWithLayout(literal.shape())); in ComputeAndCompareLiteralWithAllInputLayouts()
244 ShapeUtil::HumanStringWithLayout(literal_relayout.shape())); in ComputeAndCompareLiteralWithAllInputLayouts()
/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/kernels/xla/
H A Doutfeed_ops.cc74 << xla::ShapeUtil::HumanStringWithLayout(tuple_shape); in Compile()

123