Searched refs:param_shape_index (Results 1 – 4 of 4) sorted by relevance
257 arguments[param_no]->buffer(allocation.param_shape_index()); in ExecuteOnStream()266 allocation.param_shape_index().ToString(), param_no); in ExecuteOnStream()333 slice.allocation()->param_shape_index()); in ExecuteOnStream()337 << slice.allocation()->param_shape_index() in ExecuteOnStream()
94 allocation.param_shape_index()); in CreateBufferTable()254 slice.allocation()->param_shape_index()); in CreateResultShapedBuffer()258 << slice.allocation()->param_shape_index() in CreateResultShapedBuffer()
131 const ShapeIndex& param_shape_index() const { in param_shape_index() function281 ShapeIndex param_shape_index, in set_entry_computation_parameter() argument286 param_shape_index_ = std::move(param_shape_index); in set_entry_computation_parameter()
262 for (int64 idx : param_shape_index()) { in ToProto()292 param_shape_index().ToString()); in ToString()