| /external/tensorflow/tensorflow/core/kernels/ |
| sparse_sparse_binary_op_shared.cc | 124 const Tensor *a_indices_t, *a_values_t, *a_shape_t, *b_indices_t, variable
128 OP_REQUIRES_OK(ctx, ctx->input("a_shape", &a_shape_t));
163 TensorShapeUtils::IsVector(a_shape_t->shape()) &&
167 a_shape_t->shape().DebugString(), " and ",
169 OP_REQUIRES(ctx, a_shape_t->IsSameSize(*b_shape_t),
172 a_shape_t->SummarizeValue(10), " and ",
174 const auto a_shape = a_shape_t->flat<int64>();
176 for (int i = 0; i < a_shape_t->NumElements(); ++i) {
|
| sparse_tensor_dense_matmul_op_test.cc | 48 auto a_shape_t = a_shape.vec<int64>(); local
49 a_shape_t(0) = adjoint_a ? k : m;
50 a_shape_t(1) = adjoint_a ? m : k;
55 std::uniform_int_distribution<> a_lhs_dist(0, a_shape_t(0) - 1);
56 std::uniform_int_distribution<> a_rhs_dist(0, a_shape_t(1) - 1);
|
| sparse_tensor_dense_add_op.cc | 75 const Tensor *a_indices_t, *a_values_t, *a_shape_t, *b; variable
78 OP_REQUIRES_OK(ctx, ctx->input("a_shape", &a_shape_t));
81 ctx, ValidateInputs<Index>(a_indices_t, a_values_t, a_shape_t, b));
|
| sparse_tensor_dense_matmul_op.cc | 78 auto a_shape_t = a_shape->vec<int64>(); variable
79 const int64 outer_left = (adjoint_a_) ? a_shape_t(1) : a_shape_t(0);
82 const int64 inner_left = (adjoint_a_) ? a_shape_t(0) : a_shape_t(1);
93 a_shape_t(0), ", ", a_shape_t(1),
|
| /external/tensorflow/tensorflow/core/ops/ |
| sparse_ops_test.cc | 200 Tensor a_shape_t = test::AsTensor<int64>(std::vector<int64>{3, 1}); local
202 op.input_tensors[2] = &a_shape_t;
214 a_shape_t = test::AsTensor<int64>(std::vector<int64>{3, 1, 2});
|
