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

  /external/tensorflow/tensorflow/compiler/xla/service/
layout_assignment_test.cc 96 Shape ashape = ShapeUtil::MakeShape(F32, {42, 12}); local
98 HloInstruction::CreateParameter(0, ashape, "param0"));
100 HloInstruction::CreateParameter(1, ashape, "param1"));
102 HloInstruction::CreateBinary(ashape, HloOpcode::kAdd, param0, param1));
107 Shape shape(ashape);
126 Shape ashape = ShapeUtil::MakeShape(F32, {42, 12}); local
128 HloInstruction::CreateParameter(0, ashape, "param0"));
130 HloInstruction::CreateParameter(1, ashape, "param1"));
132 HloInstruction::CreateBinary(ashape, HloOpcode::kAdd, param0, param1));
137 Shape col_major_shape(ashape);
169 Shape ashape = constant_literal1.shape(); local
352 Shape ashape = ShapeUtil::MakeShape(F32, {1, 2, 3, 1}); local
394 Shape ashape = ShapeUtil::MakeShape(F32, {42, 12}); local
431 Shape ashape = ShapeUtil::MakeShape(F32, {3, 4}); local
546 Shape ashape = ShapeUtil::MakeShape(F32, {50, 1}); local
1325 Shape ashape = ShapeUtil::MakeShape(F32, {12, 8}); local
    [all...]
  /external/tensorflow/tensorflow/compiler/xla/service/gpu/
gpu_layout_assignment_test.cc 42 Shape ashape = ShapeUtil::MakeShape(F32, {42, 12}); local
43 Shape ashape_in_row_major(ashape);
44 Shape ashape_in_col_major(ashape);
59 HloInstruction::CreateParameter(0, ashape, "x"));
61 HloInstruction::CreateParameter(1, ashape, "y"));
63 HloInstruction::CreateBinary(ashape, HloOpcode::kAdd, x, y));
  /external/toolchain-utils/cros_utils/
pstat.py 946 ashape = a.shape
961 return N.reshape(work, ashape)
    [all...]

Completed in 128 milliseconds