Home
last modified time | relevance | path

Searched defs:param_shape_index (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dbuffer_assignment.h132 const ShapeIndex& param_shape_index() const { in param_shape_index() function
277 ShapeIndex param_shape_index, in set_entry_computation_parameter()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dgpu_compiler.cc1579 auto param_shape_index = buffer->mutable_lmhlo_param_shape_index(); in GetMlirAllocationInfo() local