Searched refs:runtime_shape (Results 1 – 4 of 4) sorted by relevance
218 const xla::Shape& runtime_shape = in UpdateDynamicInputs() local221 TF_RET_CHECK(!runtime_shape.IsTuple()); in UpdateDynamicInputs()223 DynamicShapeIsCompatible(runtime_shape, compile_time_shape)); in UpdateDynamicInputs()230 ShapeSizeCompact(runtime_shape) / sizeof(uint32)); in UpdateDynamicInputs()234 ShapeSizeCompactRaw(runtime_shape))); in UpdateDynamicInputs()236 runtime_shape, compile_time_shape]() { in UpdateDynamicInputs()240 ApiConverter::ToC(runtime_shape, &c_runtime_shape); in UpdateDynamicInputs()252 params.runtime_shape = &c_runtime_shape; in UpdateDynamicInputs()
191 XLA_Shape* runtime_shape; member
157 const xla::Shape& runtime_shape) { in UpdateMetadata() argument172 xla::LiteralUtil::CreateR1<int32>(PrepareMetadata(runtime_shape))); in UpdateMetadata()224 const xla::Shape* runtime_shape, in UpdateDynamicInputs()226 TF_RET_CHECK(!runtime_shape->IsTuple()); in UpdateDynamicInputs()228 *runtime_shape, sub_shape)); in UpdateDynamicInputs()241 sub_shape, *runtime_shape)); in UpdateDynamicInputs()
763 RuntimeShape runtime_shape = GetTensorShape(dense_key_tensor); in PrepareParseExample() local