Searched refs:c_shape (Results 1 – 4 of 4) sorted by relevance
/external/tensorflow/tensorflow/core/kernels/sparse/ |
D | mat_mul_op.cc | 524 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/ |
D | shared_device_buffer_test.cc | 76 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/ |
D | sparse_csr_matrix_ops.py | 140 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/ |
D | constant_folding.cc | 3110 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()
|