Home
last modified time | relevance | path

Searched defs:n_dim (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dconv_op_helpers.cc438 int n_dim = GetTensorBatchDimIndex(num_dims, attrs.data_format); in MakeXlaBackpropFilterConvOp() local
/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
1249 int64 n_dim = b_matrix_shape.dim(1).size(); in PredictSparseTensorDenseMatMul() local
/external/tensorflow/tensorflow/core/ops/
Dnn_ops.cc1268 DimensionHandle n_dim; in __anon3e672dd82602() local
Darray_ops.cc83 DimensionHandle n_dim = c->Dim(paddings, 0); in PadShapeFn() local