Home
last modified time | relevance | path

Searched defs:first_dim_size (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Ddynamic_stitch_op.cc77 OpInputList* data_inputs, int* first_dim_size, in CheckArgsAndAllocateResult()
167 int first_dim_size; in Compute() local
239 int first_dim_size; in Compute() local
Ddynamic_stitch_op_gpu.cu.cc53 const int32 slice_size, const int32 first_dim_size, in DynamicStitchGPUImpl()
Dtraining_ops.cc176 const Tindex first_dim_size = static_cast<Tindex>(var.dimension(0)); in operator ()() local
285 const Tindex first_dim_size = static_cast<Tindex>(var.dimension(0)); in operator ()() local
612 const Tindex first_dim_size = in operator ()() local
647 const Tindex first_dim_size = accum_flat.size(); in operator ()() local
728 const Tindex first_dim_size = static_cast<Tindex>(var.dimension(0)); in operator ()() local
1268 const Tindex first_dim_size = var.dim_size(0); in DoCompute() local
1482 const Tindex first_dim_size = var.dim_size(0); in Compute() local
1522 const Tindex first_dim_size = var_flat.size(); in Compute() local
2488 const Tindex first_dim_size = var.dim_size(0); in Compute() local
2534 const Tindex first_dim_size = var_flat.size(); in Compute() local
[all …]
Dtraining_ops_gpu.cu.cc439 const Tindex first_dim_size = var.dimension(0); in operator ()() local
500 const Tindex first_dim_size = var.dimension(0); in operator ()() local
710 const Tindex first_dim_size = var.dimension(0); in operator ()() local
786 const Tindex first_dim_size = var.dimension(0); in operator ()() local
852 const Tindex first_dim_size = var.dimension(0); in operator ()() local
/external/pytorch/aten/src/ATen/native/
DTensorShape.h25 int64_t first_dim_size = first.sizes()[dim]; in check_cat_shape_except_dim() local
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf.cc5831 int64_t first_dim_size = input_type.getDimSize(0); in matchAndRewrite() local