Searched refs:EinsumHelper (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/linalg/ |
| D | einsum_op_impl.h | 65 struct EinsumHelper { struct
595 c, EinsumHelper::ParseEquation( in EinsumOp()
607 std::vector<EinsumHelper::DimensionType> label_types(label_types_); in Compute()
612 OP_REQUIRES_OK(ctx, EinsumHelper::ProcessDimensions( in Compute()
629 EinsumHelper::ReduceOperand<Device, T>( in Compute()
639 OP_REQUIRES_OK(ctx, EinsumHelper::ContractOperands<Device, T>( in Compute()
653 if (label_types[label] == EinsumHelper::kBroadcasting) in Compute()
657 if (label_types[label] == EinsumHelper::kBatch) in Compute()
671 ctx, EinsumHelper::CopyFrom(contraction_output_reshaped, result_shape, in Compute()
680 ctx, EinsumHelper::StrideOrInflate<Device, T>( in Compute()
[all …]
|