| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | runtime_fft_impl.h | 78 Eigen::DSizes<Eigen::DenseIndex, FFTRank + 1> out_dims; in EigenFftR2C() local
79 out_dims[0] = input_batch; in EigenFftR2C()
82 out_dims[i + 1] = i == FFTRank - 1 ? fft_shape[i] / 2 + 1 : fft_shape[i]; in EigenFftR2C()
89 output(out, out_dims); in EigenFftR2C()
102 output.device(device) = full_fft.slice(zero_start_indices, out_dims); in EigenFftR2C()
117 Eigen::DSizes<Eigen::DenseIndex, FFTRank + 1> out_dims; in EigenFftC2R() local
118 out_dims[0] = input_batch; in EigenFftC2R()
121 out_dims[i + 1] = fft_shape[i]; in EigenFftC2R()
128 output(out, out_dims); in EigenFftC2R()
134 Eigen::Tensor<Complex, FFTRank + 1, Eigen::RowMajor> full_fft(out_dims); in EigenFftC2R()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | diag_op.cc | 127 const int out_dims = num_dims / 2; in Compile() local
138 for (int i = 0; i < out_dims; i++) { in Compile()
140 ctx, dims[i] == dims[i + out_dims], in Compile()
142 ": dimensions ", i, " and ", i + out_dims, in Compile()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | diag_op.cc | 80 const int out_dims = num_dims / 2; in Compute() local
85 for (int i = 0; i < out_dims; i++) { in Compute()
87 context, tensor.dim_size(i) == tensor.dim_size(i + out_dims), in Compute()
90 i, " and ", i + out_dims, " do not match.")); in Compute()
94 for (int i = 0; i < out_dims; ++i) { in Compute()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | kernel_util.cc | 411 int out_dims = std::max(dims1, dims2); in CalculateShapeForBroadcast() local
417 TfLiteIntArrayCreate(out_dims), TfLiteIntArrayFree); in CalculateShapeForBroadcast()
418 for (int i = 0; i < out_dims; ++i) { in CalculateShapeForBroadcast()
428 shape->data[out_dims - i - 1] = std::max(d1, d2); in CalculateShapeForBroadcast()
442 int out_dims = std::max(std::max(dims1, dims2), dims3); in CalculateShapeForBroadcast() local
444 TfLiteIntArrayCreate(out_dims), TfLiteIntArrayFree); in CalculateShapeForBroadcast()
445 for (int i = 0; i < out_dims; ++i) { in CalculateShapeForBroadcast()
459 shape->data[out_dims - i - 1] = max_value; in CalculateShapeForBroadcast()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | reference_util.h | 255 int64 out_dims[] = {rhs.n1(), rhs.n2(), rhs.n3()}; in Concat3D() local
258 out_dims[i] = lhs_dims[i]; in Concat3D()
261 out_dims[i] = lhs_dims[i] + rhs_dims[i]; in Concat3D()
265 absl::make_unique<Array3D<T>>(out_dims[0], out_dims[1], out_dims[2]); in Concat3D()
290 int64 out_dims[] = {rhs.n1(), rhs.n2(), rhs.n3(), rhs.n4()}; in Concat4D() local
293 out_dims[i] = lhs_dims[i]; in Concat4D()
296 out_dims[i] = lhs_dims[i] + rhs_dims[i]; in Concat4D()
299 auto result = absl::make_unique<Array4D<T>>(out_dims[0], out_dims[1], in Concat4D()
300 out_dims[2], out_dims[3]); in Concat4D()
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_batch_matmul_op.cc | 185 auto out_dims = TFShapeToMklDnnDims(out_shape); in CreateMatMulParams() local
205 auto out_strides = CalculateTFStrides(out_dims); in CreateMatMulParams()
227 lhs_dims, rhs_dims, out_dims, lhs_strides, rhs_strides, out_strides); in CreateMatMulParams()
|
| D | mkl_transpose_op.cc | 133 memory::dims out_dims = TFShapeToMklDnnDims(out_tensor->shape()); in MKLTransposeND() local
137 ReorderStrides(CalculateTFStrides(out_dims), perm); in MKLTransposeND()
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | gradient_checker.cc | 39 void GetDims(const TF_Tensor* t, int64_t* out_dims) { in GetDims() argument
42 out_dims[i] = TF_Dim(t, i); in GetDims()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | tensor.cc | 1272 gtl::InlinedVector<int64, 4> out_dims(num_out_dims, 0); in ComputeFlatInnerDims() local
1276 out_dims[out_dim] = in_dim < 0 ? 1 : orig[in_dim]; in ComputeFlatInnerDims()
1279 out_dims[0] *= orig[in_dim]; in ComputeFlatInnerDims()
1281 return out_dims; in ComputeFlatInnerDims()
1286 gtl::InlinedVector<int64, 4> out_dims(num_out_dims, 0); in ComputeFlatOuterDims() local
1288 out_dims[out_dim] = out_dim >= orig.size() ? 1 : orig[out_dim]; in ComputeFlatOuterDims()
1291 out_dims[num_out_dims - 1] *= orig[in_dim]; in ComputeFlatOuterDims()
1293 return out_dims; in ComputeFlatOuterDims()
|
| D | common_shape_fns.cc | 549 std::vector<DimensionHandle> out_dims(rank); in ShapeFromDimensions() local
552 out_dims[tensorflow::GetTensorBatchDimIndex(rank, format)] = batch_dim; in ShapeFromDimensions()
556 out_dims[tensorflow::GetTensorSpatialDimIndex( in ShapeFromDimensions()
565 &out_dims[tensorflow::GetTensorFeatureDimIndex(rank, format)])); in ShapeFromDimensions()
566 out_dims[GetTensorInnerFeatureDimIndex(rank, format)] = context->MakeDim(4); in ShapeFromDimensions()
568 out_dims[tensorflow::GetTensorFeatureDimIndex(rank, format)] = filter_dim; in ShapeFromDimensions()
571 *shape = context->MakeShape(out_dims); in ShapeFromDimensions()
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | self_adjoint_eig_test.cc | 105 absl::Span<const int64> out_dims = shape.dimensions(); in ComputeMatmulVWVt() local
110 auto vw = Mul(result.v, BroadcastInDim(result.w, out_dims, broadcast_dims)); in ComputeMatmulVWVt()
|
| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | local.py | 802 out_dims = shape[split_dim:]
805 out_size = math_ops.reduce_prod(out_dims)
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | tooling_util.cc | 1465 std::vector<int>* out_dims) { in MakeArrayDims() argument
1466 CHECK(out_dims->empty()); in MakeArrayDims()
1471 *out_dims = {depth}; in MakeArrayDims()
1473 *out_dims = {batch, depth}; in MakeArrayDims()
1476 *out_dims = {height, width, depth}; in MakeArrayDims()
1478 *out_dims = {batch, height, width, depth}; in MakeArrayDims()
|
| D | tooling_util.h | 262 std::vector<int>* out_dims);
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | randomized_tests.cc | 2692 std::vector<int64> out_dims = RandomDims(2, 2); in TEST_F() local
2698 std::vector<int32>(out_dims.begin(), out_dims.end()))) in TEST_F()
2707 std::vector<int64> out_dims = RandomDims(2, 2); in TEST_F() local
2713 {in_dims[0], out_dims[0], out_dims[1], in_dims[3]}) in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 1379 const int64_t out_dims = num_dims / 2; in matchAndRewrite() local
1383 for (int i = 0; i < out_dims; i++) { in matchAndRewrite()
1384 if (input_type.getDimSize(i) != input_type.getDimSize(i + out_dims)) in matchAndRewrite()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes_test.cc | 1688 auto out_dims = std::vector<int>(exp_out_dims[i].begin() + 1, in RunValidationAndConversion() local
1691 ExpectTrtDimsEqualsArray(out_dims, in RunValidationAndConversion()
|