Searched refs:param_shape_index (Results 1 – 5 of 5) sorted by relevance
316 allocation.param_shape_index()); in BufferForAllocation()319 .Buffer(allocation.param_shape_index()) in BufferForAllocation()329 allocation.param_shape_index().ToString(), param_no); in BufferForAllocation()483 return input.MutableBuffer(allocation->param_shape_index()); in ExecuteAsyncOnStreamImpl()
134 const ShapeIndex& param_shape_index() const { in param_shape_index() function278 ShapeIndex param_shape_index, in set_entry_computation_parameter() argument283 param_shape_index_ = std::move(param_shape_index); in set_entry_computation_parameter()
279 for (int64 idx : param_shape_index()) { in ToProto()348 "| at ShapeIndex ", param_shape_index().ToString()); in ToString()
98 .Buffer(allocation.param_shape_index()) in MemoryForAllocation()102 << " at shape index " << allocation.param_shape_index().ToString(); in MemoryForAllocation()
1468 lhs->parameter_number(), lhs->param_shape_index()) < in Initialize()1470 rhs->parameter_number(), rhs->param_shape_index()); in Initialize()1489 alloc->param_shape_index()); in Initialize()