Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/tpu/kernels/xla/
Dinplace_ops.cc46 const TensorShape i_shape = ctx->InputShape(1); in Compile() local
106 const TensorShape i_shape = ctx->InputShape(1); in Compile() local
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dbincount_op.cc109 auto i_shape = in Compile() local
/external/ComputeLibrary/src/cpu/operators/
DCpuGemmDirectConv2d.cpp168 const TensorShape i_shape = src->tensor_shape(); in validate() local
/external/tensorflow/tensorflow/core/common_runtime/
Dfunction_test.cc1893 auto i_shape = in TEST_F() local
/external/pytorch/test/
Dtest_sparse.py2794 def test_shape(i_shape, v_shape, size, expected_size): argument
/external/tensorflow/tensorflow/compiler/tests/
Drandomized_tests.cc2991 std::vector<int64_t> i_shape{v_dims[0]}; in TEST_F() local