Home
last modified time | relevance | path

Searched refs:GetForwardedDTensorLayoutInput (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/
H A Dspmd_expander_common.cc321 mlir::Value GetForwardedDTensorLayoutInput(mlir::Value value) { in GetForwardedDTensorLayoutInput() function
682 value = GetForwardedDTensorLayoutInput(value); in ExtractConstStringVectorFromValue()
699 value = GetForwardedDTensorLayoutInput(value); in ExtractConstScalarStringFromValue()
H A Dspmd_expander_common.h90 mlir::Value GetForwardedDTensorLayoutInput(mlir::Value value);
H A Dspmd_expansion.cc182 GetForwardedDTensorLayoutInput(assign_variable_op.resource()) in IsValueUsedByAssignVariableOp()
/aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/expansions/
H A Dbroadcast_to_spmd_expander.cc59 GetForwardedDTensorLayoutInput(broadcast_op.shape()), in ExpandOp()