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

  /external/tensorflow/tensorflow/compiler/xla/service/
logical_buffer_analysis.cc 101 ShapeUtil::ForEachSubshape(
allocation_tracker.cc 55 ShapeUtil::ForEachSubshape(shaped_buffer->on_device_shape(),
80 ShapeUtil::ForEachSubshape(shaped_buffer->on_device_shape(),
shaped_buffer.cc 85 ShapeUtil::ForEachSubshape(
hlo_cost_analysis.cc 421 ShapeUtil::ForEachSubshape(
455 ShapeUtil::ForEachSubshape(
instruction_fusion.cc 136 ShapeUtil::ForEachSubshape(
buffer_assignment.cc     [all...]
hlo_alias_analysis.cc 383 ShapeUtil::ForEachSubshape(
copy_insertion.cc     [all...]
hlo_rematerialization.cc     [all...]
buffer_assignment_test.cc     [all...]
hlo_graph_dumper.cc     [all...]
  /external/tensorflow/tensorflow/compiler/xla/service/gpu/
hlo_to_ir_bindings.cc 94 ShapeUtil::ForEachSubshape(
ir_emitter_unnested.cc     [all...]
  /external/tensorflow/tensorflow/compiler/xla/
shape_util.h 447 static void ForEachSubshape(const Shape& shape, const VisitorFunction& func);
shape_util_test.cc 496 ShapeUtil::ForEachSubshape(
511 ShapeUtil::ForEachSubshape(
shape_util.cc 897 // Helper for ForEachSubshape which visits the subshapes of the given shape in
933 /* static */ void ShapeUtil::ForEachSubshape(const Shape& shape,
    [all...]
literal_util.cc 724 ShapeUtil::ForEachSubshape(
    [all...]
  /external/tensorflow/tensorflow/compiler/xla/tests/
literal_test_util.cc 124 ShapeUtil::ForEachSubshape(

Completed in 428 milliseconds