Searched refs:arg_tensor_shape (Results 1 – 1 of 1) sorted by relevance
1451 tensorflow::TensorShape arg_tensor_shape; in SetEntryTupleShapesAndLeafReplication() local1453 tensorflow::XLAShapeToTensorShape(arg_shape, &arg_tensor_shape); in SetEntryTupleShapesAndLeafReplication()1462 auto arg_shape_status = shape_representation_fn_(arg_tensor_shape, dtype, in SetEntryTupleShapesAndLeafReplication()