Searched refs:shape_in (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | xent_op.cc | 44 TensorShape shape_in = logits_in.shape(); in Compute() local
54 shape_in = BCast::ToShape(bcast.output_shape()); in Compute()
56 OP_REQUIRES(context, TensorShapeUtils::IsMatrix(shape_in), in Compute()
66 TensorShape({shape_in.dim_size(0), 1}), in Compute()
72 0, TensorShape({shape_in.dim_size(0)}), &loss_out)); in Compute()
76 {0}, 1, shape_in, &back_out)); in Compute()
77 if (shape_in.dim_size(0) > 0) { in Compute()
80 functor(context->eigen_device<Device>(), shape_in.AsEigenDSizes<2>(), in Compute()
86 functor(context->eigen_device<Device>(), shape_in.AsEigenDSizes<2>(), in Compute()
|
| /external/python/pybind11/include/pybind11/ |
| D | buffer_info.h | 53 …detail::any_container<ssize_t> shape_in, detail::any_container<ssize_t> strides_in, bool readonly=…
55 shape(std::move(shape_in)), strides(std::move(strides_in)), readonly(readonly) { in ptr()
63 …buffer_info(T *ptr, detail::any_container<ssize_t> shape_in, detail::any_container<ssize_t> stride…
64 …(T), format_descriptor<T>::format(), static_cast<ssize_t>(shape_in->size()), std::move(shape_in), … in private_ctr_tag()
122 …detail::any_container<ssize_t> &&shape_in, detail::any_container<ssize_t> &&strides_in, bool reado… in buffer_info()
123 …: buffer_info(ptr, itemsize, format, ndim, std::move(shape_in), std::move(strides_in), readonly) {… in buffer_info()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | shape_inference_test.cc | 866 Shape shape_in = ShapeUtil::MakeShape(F32, {16, 8}); in TEST_F() local
868 fft::Fail(shape_in, type, {}, fft::unsupported_rank); in TEST_F()
869 fft::Pass(shape_in, type, {8}, shape_out); in TEST_F()
870 fft::Pass(shape_in, type, {16, 8}, shape_out); in TEST_F()
871 fft::Fail(shape_in, type, {32, 16, 8}, fft::invalid_rank); in TEST_F()
872 fft::Fail(shape_in, type, {64, 32, 16, 8}, fft::unsupported_rank); in TEST_F()
905 Shape shape_in = ShapeUtil::MakeShape(C64, {16, 5}); in TEST_F() local
907 fft::Fail(shape_in, type, {}, fft::unsupported_rank); in TEST_F()
908 fft::Pass(shape_in, type, {8}, shape_out); in TEST_F()
909 fft::Pass(shape_in, type, {16, 8}, shape_out); in TEST_F()
[all …]
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | graph_properties_test.cc | 619 auto shape_in = properties.GetOutputProperties("ones").at(0).shape(); in TEST_F() local
621 EXPECT_GE(-2, shape_in.dim(0).size()); in TEST_F()
623 EXPECT_NE(shape_in.dim(0).size(), shape_out.dim(0).size()); in TEST_F()
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | array_ops.cc | 475 ShapeHandle shape_in = c->input(1); in __anondb9326b20702() local
476 TF_RETURN_IF_ERROR(c->WithRank(shape_in, 1, &shape_in)); in __anondb9326b20702()
|