Searched refs:shape_size_fn (Results 1 – 2 of 2) sorted by relevance
163 auto shape_size_fn = compiler->ShapeSizeBytesFunction(); in UpdateMetadata() local166 uint64 offset = shape_size_fn(compile_time_shape_static); in UpdateMetadata()167 uint64 metadata_size = shape_size_fn(compile_time_shape) - offset; in UpdateMetadata()208 auto shape_size_fn = compiler->ShapeSizeBytesFunction(); in UpdateDynamicInputs() local232 shape_size_fn(sub_shape))); in UpdateDynamicInputs()
225 auto shape_size_fn = compiler->ShapeSizeBytesFunction(); in ReadDynamicShapes() local227 const int64 offset = shape_size_fn(buffer_shape_static); in ReadDynamicShapes()228 int64 metadata_size = shape_size_fn(buffer_shape) - offset; in ReadDynamicShapes()