Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/tpu/kernels/
Dsharding_util_ops.cc370 const auto& ishape = input->shape(); in ValidateShapesForSlice() local
/external/pytorch/aten/src/ATen/native/cpu/
DPaddingKernel.cpp24 c10::SmallVector<int64_t, 3u> ishape; member
/external/tensorflow/tensorflow/python/kernel_tests/nn_ops/
Dconv_ops_test.py3479 ishape = [1, 400, 400, 1] variable
/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Ddistributed_tpu_rewrite_pass.cc498 static int64_t GetInferredShapeSize(const InferredShape& ishape, in GetInferredShapeSize()