Home
last modified time | relevance | path

Searched defs:keys_shape (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_layout_assignment.cc220 Shape keys_shape = instruction->operand(0)->shape(); in AddBackendConstraints() local
Dir_emitter_unnested.cc1211 Shape keys_shape = sort->operand(0)->shape(); in HandleSort() local
/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Dsort_util.cc307 const Shape& keys_shape = values_arrays[0].GetShape(); in EmitSortInPlace() local
/external/tensorflow/tensorflow/compiler/xla/service/
Dtuple_points_to_analysis_test.cc1071 Shape keys_shape = ShapeUtil::MakeShape(F32, {8}); in TEST_F() local
1089 Shape keys_shape = ShapeUtil::MakeShape(F32, {8}); in TEST_F() local
Dhlo_dataflow_analysis_test.cc2450 Shape keys_shape = ShapeUtil::MakeShape(F32, {8}); in TEST_F() local
2468 Shape keys_shape = ShapeUtil::MakeShape(F32, {8}); in TEST_F() local
Dalgebraic_simplifier_test.cc2804 Shape keys_shape = ShapeUtil::MakeShape(F32, {1}); in TEST_F() local
2820 Shape keys_shape = ShapeUtil::MakeShape(F32, {5, 0}); in TEST_F() local
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.cc507 Shape keys_shape = sort->keys()->shape(); in HandleSort() local