Home
last modified time | relevance | path

Searched refs:c_dim (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dconv_op_helpers.cc413 int c_dim = GetTensorFeatureDimIndex(num_dims, attrs.data_format); in MakeXlaBackpropFilterConvOp() local
414 int64 in_depth = input_shape.dimensions(c_dim), in MakeXlaBackpropFilterConvOp()
425 dnums.set_input_batch_dimension(c_dim); in MakeXlaBackpropFilterConvOp()
432 dnums.set_kernel_output_feature_dimension(c_dim); in MakeXlaBackpropFilterConvOp()
/external/tensorflow/tensorflow/compiler/xla/service/mlir_gpu/experimental/conv_emitter/
Dconv_emitter.cc62 const Shape& shape, int64 n_dim, int64 c_dim, in GetShapeInfo() argument
74 nchw_to_logical.push_back(c_dim); in GetShapeInfo()