Home
last modified time | relevance | path

Searched refs:n_dim (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/costs/
Dop_level_cost_estimator.cc704 double m_dim, n_dim, k_dim, k_dim_b = 0; in CountMatMulOperations() local
729 n_dim = b_matrix_shape.dim(0).size(); in CountMatMulOperations()
732 n_dim = b_matrix_shape.dim(1).size(); in CountMatMulOperations()
735 VLOG(1) << "M, N, K: " << m_dim << "," << n_dim << "," << k_dim; in CountMatMulOperations()
746 ops = m_dim * n_dim * k_dim * 2; in CountMatMulOperations()
751 mat_mul->n = n_dim; in CountMatMulOperations()
1249 int64 n_dim = b_matrix_shape.dim(1).size(); in PredictSparseTensorDenseMatMul() local
1253 const int64 op_count = kOpsPerMac * num_elems_in_a * n_dim; in PredictSparseTensorDenseMatMul()
1262 num_elems_in_a * n_dim * DataTypeSize(BaseType(b_matrix.dtype())); in PredictSparseTensorDenseMatMul()
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dconv_op_helpers.cc438 int n_dim = GetTensorBatchDimIndex(num_dims, attrs.data_format); in MakeXlaBackpropFilterConvOp() local
451 dnums.set_input_feature_dimension(n_dim); in MakeXlaBackpropFilterConvOp()
456 dnums.set_kernel_input_feature_dimension(n_dim); in MakeXlaBackpropFilterConvOp()
/external/tensorflow/tensorflow/core/ops/
Darray_ops.cc83 DimensionHandle n_dim = c->Dim(paddings, 0); in PadShapeFn() local
84 if (c->ValueKnown(n_dim)) { in PadShapeFn()
85 TF_RETURN_IF_ERROR(c->WithRank(input, c->Value(n_dim), &input)); in PadShapeFn()
87 TF_RETURN_IF_ERROR(c->WithValue(n_dim, c->Rank(input), &n_dim)); in PadShapeFn()
94 if (c->ValueKnown(n_dim)) { in PadShapeFn()
96 c->set_output(0, c->UnknownShapeOfRank(c->Value(n_dim))); in PadShapeFn()
105 TF_RETURN_IF_ERROR(c->WithValue(n_dim, num_dims, &n_dim)); in PadShapeFn()
Dnn_ops.cc1268 DimensionHandle n_dim; in __anon3e672dd82602() local
1269 TF_RETURN_IF_ERROR(c->MakeDimForScalarInput(1, &n_dim)); in __anon3e672dd82602()
1273 if (c->ValueKnown(last_dim) && c->ValueKnown(n_dim) && in __anon3e672dd82602()
1274 c->Value(last_dim) <= c->Value(n_dim)) { in __anon3e672dd82602()
1276 "Input must have last dimension > n = ", c->Value(n_dim), in __anon3e672dd82602()
/external/tensorflow/tensorflow/contrib/rnn/python/ops/
Drnn_cell.py1824 n_dim = i_dim = self._num_units
1827 self._linear2 = _Linear(cell_inputs, 2 * n_dim + 2 * i_dim, True)
1830 gh_act = rnn_matrix[:, :n_dim] # b x n
1831 h_act = rnn_matrix[:, n_dim:2 * n_dim] # b x n
1832 gy_act = rnn_matrix[:, 2 * n_dim:2 * n_dim + i_dim] # b x i
1833 y_act = rnn_matrix[:, 2 * n_dim + i_dim:2 * n_dim + 2 * i_dim] # b x i
/external/tensorflow/tensorflow/python/kernel_tests/
Dtensor_array_ops_test.py1651 n_dim = 1
1656 dtypes.float32, size=n_time, element_shape=[n_dim])