Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/xla/experimental/conv_emitter/
Dconv_emitter.cc69 const Shape& shape, int64_t n_dim, int64_t c_dim, in GetShapeInfo()
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dconv_op_helpers.cc430 int n_dim = GetTensorBatchDimIndex(num_dims, attrs.data_format); in MakeXlaBackpropFilterConvOp() local
/external/tensorflow/tensorflow/core/grappler/costs/
Dop_level_cost_estimator.cc1069 double m_dim, n_dim, k_dim, k_dim_b = 0; in CountMatMulOperations() local
1196 TensorShapeProto_Dim n_dim; in GenerateBatchMatmulContextFromEinsum() local
1874 int64_t n_dim = b_matrix_shape.dim(1).size(); in PredictSparseTensorDenseMatMul() local
/external/tensorflow/tensorflow/core/ops/
Dnn_ops.cc1491 DimensionHandle n_dim; in __anon1bbab8942102() local
Darray_ops.cc91 DimensionHandle n_dim = c->Dim(paddings, 0); in PadShapeFn() local