Searched refs:n_dim (Results 1 – 6 of 6) sorted by relevance
/external/tensorflow/tensorflow/core/grappler/costs/ |
D | op_level_cost_estimator.cc | 704 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/ |
D | conv_op_helpers.cc | 438 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/ |
D | array_ops.cc | 83 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()
|
D | nn_ops.cc | 1268 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/ |
D | rnn_cell.py | 1824 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/ |
D | tensor_array_ops_test.py | 1651 n_dim = 1 1656 dtypes.float32, size=n_time, element_shape=[n_dim])
|