Searched refs:arg_shape_status (Results 1 – 1 of 1) sorted by relevance
1462 auto arg_shape_status = shape_representation_fn_(arg_tensor_shape, dtype, in SetEntryTupleShapesAndLeafReplication() local1464 if (!arg_shape_status.ok()) in SetEntryTupleShapesAndLeafReplication()1466 << arg_shape_status.status().error_message(); in SetEntryTupleShapesAndLeafReplication()1468 arg_shape = std::move(arg_shape_status.ValueOrDie()); in SetEntryTupleShapesAndLeafReplication()