| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_batch_matmul_op.cc | 183 auto lhs_dims = TFShapeToMklDnnDims(lhs_shape); in CreateMatMulParams() local
193 ExpandInputDimsToOutputShape(lhs_shape, out_shape, &lhs_dims); in CreateMatMulParams()
203 auto lhs_strides = CalculateTFStrides(lhs_dims); in CreateMatMulParams()
210 m = lhs_dims[m_idx]; in CreateMatMulParams()
211 k = lhs_dims[k_idx]; in CreateMatMulParams()
212 std::swap(lhs_dims[m_idx], lhs_dims[k_idx]); in CreateMatMulParams()
227 lhs_dims, rhs_dims, out_dims, lhs_strides, rhs_strides, out_strides); in CreateMatMulParams()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | sparse_dense_binary_op_shared.cc | 84 const auto lhs_dims = BCast::FromShape(TensorShape(shape_vec)); in Compute() local
86 BCast b(lhs_dims, rhs_dims, false); // false for keeping the same num dims. in Compute()
97 OP_REQUIRES(ctx, VecGreaterEq(lhs_dims, rhs_dims) && b.IsValid(), in Compute()
101 absl::StrJoin(lhs_dims, ","), "] vs. [", in Compute()
120 const int ndims = lhs_dims.size(); in Compute()
|
| /external/eigen/unsupported/Eigen/CXX11/src/Tensor/ |
| D | TensorConcatenation.h | 135 const Dimensions& lhs_dims = m_leftImpl.dimensions();
140 eigen_assert(lhs_dims[i] > 0);
141 eigen_assert(lhs_dims[i] == rhs_dims[i]);
142 m_dimensions[i] = lhs_dims[i];
144 eigen_assert(lhs_dims[i] > 0); // Now i == m_axis.
146 m_dimensions[i] = lhs_dims[i] + rhs_dims[i];
148 eigen_assert(lhs_dims[i] > 0);
149 eigen_assert(lhs_dims[i] == rhs_dims[i]);
150 m_dimensions[i] = lhs_dims[i];
160 m_leftStrides[j] = m_leftStrides[j-1] * lhs_dims[j-1];
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | reference_util.h | 253 const int64 lhs_dims[] = {lhs.n1(), lhs.n2(), lhs.n3()}; in Concat3D() local
258 out_dims[i] = lhs_dims[i]; in Concat3D()
259 CHECK_EQ(lhs_dims[i], rhs_dims[i]); in Concat3D()
261 out_dims[i] = lhs_dims[i] + rhs_dims[i]; in Concat3D()
288 const int64 lhs_dims[] = {lhs.n1(), lhs.n2(), lhs.n3(), lhs.n4()}; in Concat4D() local
293 out_dims[i] = lhs_dims[i]; in Concat4D()
294 CHECK_EQ(lhs_dims[i], rhs_dims[i]); in Concat4D()
296 out_dims[i] = lhs_dims[i] + rhs_dims[i]; in Concat4D()
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | xla_client.py | 389 def window_padding_type_to_pad_values(padding_type, lhs_dims, rhs_dims, argument
408 out_shape = np.ceil(np.true_divide(lhs_dims, window_strides)).astype(int)
412 out_shape, window_strides, rhs_dims, lhs_dims)
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | dot_handler.cc | 1669 std::vector<int64> lhs_dims; in PartitionDotGroupOnBatch() local
1684 lhs_dims.push_back(dim.lhs); in PartitionDotGroupOnBatch()
1702 GroupShardingOnDims(lhs.sharding(), lhs_dims))); in PartitionDotGroupOnBatch()
1708 auto lhs_grouped = GroupShardingOnDims(lhs.sharding(), lhs_dims); in PartitionDotGroupOnBatch()
1846 lhs, lhs_dims, lhs_contracting_dims, lhs_non_contracting_dims, in PartitionDotGroupOnBatch()
2102 std::vector<int64> lhs_dims; in PartitionDotGroupOnContracting() local
2106 lhs_dims.push_back(dim.lhs); in PartitionDotGroupOnContracting()
2130 auto lhs_grouped = GroupShardingOnDims(lhs_sharding, lhs_dims); in PartitionDotGroupOnContracting()
2151 if (absl::c_linear_search(lhs_dims, i)) { in PartitionDotGroupOnContracting()
2238 b, dot, lhs_sharding, lhs.state().next_channel_id, lhs_dims, in PartitionDotGroupOnContracting()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_instruction.cc | 3708 std::vector<string> lhs_dims(2 + dnums.input_spatial_dimensions().size()); in ConvolutionDimensionNumbersToString() local
3709 lhs_dims[dnums.input_batch_dimension()] = 'b'; in ConvolutionDimensionNumbersToString()
3710 lhs_dims[dnums.input_feature_dimension()] = 'f'; in ConvolutionDimensionNumbersToString()
3712 lhs_dims[dnums.input_spatial_dimensions(i)] = StrCat(i); in ConvolutionDimensionNumbersToString()
3729 return StrCat(StrJoin(lhs_dims, ""), "_", StrJoin(rhs_dims, ""), "->", in ConvolutionDimensionNumbersToString()
|
| D | elemental_ir_emitter.cc | 2183 int64 lhs_dims = hlo->operand(0)->shape().dimensions_size(); in EmitElementalDot() local
2212 for (int64 i = 0; i < lhs_dims - 1; i++) { in EmitElementalDot()
2226 rhs_multi_index.push_back(dot_result_index[lhs_dims - 1 + i]); in EmitElementalDot()
|
| D | algebraic_simplifier_test.cc | 5177 std::vector<int64> lhs_dims = {2, 3, 5}; in TEST_P() local
5178 std::vector<int64> rhs_dims = lhs_dims; in TEST_P()
5179 std::vector<int64> output_dims = lhs_dims; in TEST_P()
5181 lhs_dims.push_back(m); in TEST_P()
5185 lhs_dims.push_back(k); in TEST_P()
5193 Shape lhs_shape = ShapeUtil::MakeShape(element_type, lhs_dims); in TEST_P()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_ops_a_m.cc | 1259 std::array<int64_t, 1> lhs_dims{static_cast<int64_t>(op.values().size())}; in GetHoistParams() local
1260 auto lhs_type = RankedTensorType::get(lhs_dims, ranked.getElementType()); in GetHoistParams()
|