Searched refs:n_dim (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | op_level_cost_estimator.cc | 1057 double m_dim, n_dim, k_dim, k_dim_b = 0; in CountMatMulOperations() local
1082 n_dim = b_matrix_shape.dim(0).size(); in CountMatMulOperations()
1085 n_dim = b_matrix_shape.dim(1).size(); in CountMatMulOperations()
1088 VLOG(1) << "M, N, K: " << m_dim << "," << n_dim << "," << k_dim; in CountMatMulOperations()
1099 ops = m_dim * n_dim * k_dim * 2; in CountMatMulOperations()
1104 mat_mul->n = n_dim; in CountMatMulOperations()
1184 TensorShapeProto_Dim n_dim; in GenerateBatchMatmulContextFromEinsum() local
1188 n_dim.set_size(1); in GenerateBatchMatmulContextFromEinsum()
1219 n_dim.set_size(n_dim.size() * b_input_shape.dim(i_idx).size()); in GenerateBatchMatmulContextFromEinsum()
1227 *(b_matrix_shape->add_dim()) = n_dim; in GenerateBatchMatmulContextFromEinsum()
[all …]
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | conv_op_helpers.cc | 430 int n_dim = GetTensorBatchDimIndex(num_dims, attrs.data_format); in MakeXlaBackpropFilterConvOp() local
444 dnums.set_input_feature_dimension(n_dim); in MakeXlaBackpropFilterConvOp()
449 dnums.set_kernel_input_feature_dimension(n_dim); in MakeXlaBackpropFilterConvOp()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/experimental/conv_emitter/ |
| D | conv_emitter.cc | 67 const Shape& shape, int64 n_dim, int64 c_dim, in GetShapeInfo() argument
78 nchw_to_logical.push_back(n_dim); in GetShapeInfo()
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | array_ops.cc | 90 DimensionHandle n_dim = c->Dim(paddings, 0); in PadShapeFn() local
91 if (c->ValueKnown(n_dim)) { in PadShapeFn()
92 TF_RETURN_IF_ERROR(c->WithRank(input, c->Value(n_dim), &input)); in PadShapeFn()
94 TF_RETURN_IF_ERROR(c->WithValue(n_dim, c->Rank(input), &n_dim)); in PadShapeFn()
101 if (c->ValueKnown(n_dim)) { in PadShapeFn()
103 c->set_output(0, c->UnknownShapeOfRank(c->Value(n_dim))); in PadShapeFn()
112 TF_RETURN_IF_ERROR(c->WithValue(n_dim, num_dims, &n_dim)); in PadShapeFn()
|
| D | nn_ops.cc | 1362 DimensionHandle n_dim; in __anonad10f3362002() local
1363 TF_RETURN_IF_ERROR(c->MakeDimForScalarInput(1, &n_dim)); in __anonad10f3362002()
1367 if (c->ValueKnown(last_dim) && c->ValueKnown(n_dim) && in __anonad10f3362002()
1368 c->Value(last_dim) <= c->Value(n_dim)) { in __anonad10f3362002()
1370 "Input must have last dimension > n = ", c->Value(n_dim), in __anonad10f3362002()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf_patterns.td | 416 (HLO_ConstOp:$n_dim (GetDimensionSizeFromEnd<"0"> $input, $num_upper)),
430 $n_dim,
|
| /external/llvm-project/polly/lib/Analysis/ |
| D | ScopBuilder.cpp | 232 for (unsigned u = 0, e = S.n_dim(); u < e; u++) in partitionSetParts()
235 unsigned NumDimsS = S.n_dim(); in partitionSetParts()
330 int NumDim = Dom.n_dim(); in adjustDomainDimensions()
3352 Domain = Domain.project_out(isl::dim::set, 0, Domain.n_dim()); in getAccessDomain()
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | tensor_array_ops_test.py | 1762 n_dim = 1
1767 dtypes.float32, size=n_time, element_shape=[n_dim])
|
| /external/llvm-project/polly/lib/External/isl/include/isl/ |
| D | isl-noexceptions.h | 854 inline unsigned int n_dim() const;
2741 inline unsigned int n_dim() const;
6618 unsigned int basic_set::n_dim() const in n_dim() function
15083 unsigned int set::n_dim() const in n_dim() function
|