Searched refs:compile_time_shape (Results 1 – 3 of 3) sorted by relevance
212 [&](const xla::Shape& compile_time_shape, in UpdateDynamicInputs()214 if (compile_time_shape.IsTuple() || compile_time_shape.is_static()) { in UpdateDynamicInputs()223 DynamicShapeIsCompatible(runtime_shape, compile_time_shape)); in UpdateDynamicInputs()228 ShapeSizeCompact(compile_time_shape), -1); in UpdateDynamicInputs()236 runtime_shape, compile_time_shape]() { in UpdateDynamicInputs()241 ApiConverter::ToC(compile_time_shape, &c_compile_time_shape); in UpdateDynamicInputs()253 params.compile_time_shape = &c_compile_time_shape; in UpdateDynamicInputs()267 ShapeSizeCompact(compile_time_shape))); in UpdateDynamicInputs()
192 XLA_Shape* compile_time_shape; member
156 const xla::Shape& compile_time_shape, in UpdateMetadata() argument165 xla::ShapeUtil::MakeStaticShape(compile_time_shape); in UpdateMetadata()167 uint64 metadata_size = shape_size_fn(compile_time_shape) - offset; in UpdateMetadata()210 const xla::Shape& compile_time_shape = compile_time_shapes[i].shape(); in UpdateDynamicInputs() local211 if (compile_time_shape.is_static()) { in UpdateDynamicInputs()217 compile_time_shape, in UpdateDynamicInputs()251 TF_RETURN_IF_ERROR(execution_input->SetDynamicShape(compile_time_shape)); in UpdateDynamicInputs()