Home
last modified time | relevance | path

Searched refs:c_shape (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/sparse/
Dmat_mul_op.cc524 TensorShape c_shape; in Compute() local
525 if (rank == 3) c_shape.AddDim(batch_size); in Compute()
527 c_shape.AddDim(b_outer_dim); in Compute()
528 c_shape.AddDim(a_outer_dim); in Compute()
530 c_shape.AddDim(a_outer_dim); in Compute()
531 c_shape.AddDim(b_outer_dim); in Compute()
534 const int64 c_matrix_lhs = c_shape.dim_size(row_dim); in Compute()
535 const int64 c_matrix_rhs = c_shape.dim_size(row_dim + 1); in Compute()
538 OP_REQUIRES_OK(ctx, ctx->allocate_output(0, c_shape, &c_t)); in Compute()
/external/tensorflow/tensorflow/compiler/xla/python/
Dshared_device_buffer_test.cc76 Shape c_shape = ShapeUtil::MakeShape(S64, {}); in TEST() local
77 Shape abc_tuple_shape = ShapeUtil::MakeTupleShape({c_shape, ab_tuple_shape}); in TEST()
94 c_shape, client->backend().transfer_manager(), in TEST()
/external/tensorflow/tensorflow/python/ops/linalg/sparse/
Dsparse_csr_matrix_ops.py140 c_shape = tensor_shape.TensorShape(a_shape)
141 c_shape = tensor_shape.TensorShape(c_shape[:rank - 2] + [c_rows, c_cols])
142 c_handle = _create_handle_data_proto(c_shape.as_proto(),
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dconstant_folding.cc3110 const PartialTensorShape c_shape( in ConstantPushDown() local
3115 if (c_shape.IsFullyDefined() && x_shape.IsFullyDefined() && in ConstantPushDown()
3116 c_shape.num_elements() > x_shape.num_elements()) { in ConstantPushDown()
3118 } else if (!c_shape.unknown_rank() && !x_shape.unknown_rank() && in ConstantPushDown()
3119 c_shape.dims() > 0) { in ConstantPushDown()
3120 for (int idx = 0; idx < std::min(x_shape.dims(), c_shape.dims()); ++idx) { in ConstantPushDown()
3122 c_shape.dim_size(idx) > x_shape.dim_size(idx)) { in ConstantPushDown()